support BLOOM models

2023-05-31 16:54:06 +08:00
parent a72492e649
commit 740a5daf56
16 changed files with 134 additions and 90 deletions
--- a/src/export_model.py
+++ b/src/export_model.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Exports the fine-tuned LLaMA model.
+# Exports the fine-tuned model.
 # Usage: python export_model.py --checkpoint_dir path_to_checkpoint --output_dir path_to_save_model