Update generate_chat_bminf.py

caowanxia123 · Jun 10, 2023 · 648ad8d · 648ad8d
1 parent 2e5bf29
commit 648ad8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/Aquila/Aquila-chat/generate_chat_bminf.py b/examples/Aquila/Aquila-chat/generate_chat_bminf.py
@@ -6,7 +6,7 @@
 from flagai.data.tokenizer import Tokenizer
 import bminf
 
-state_dict = "/data2/yzd/checkpoints/converted_models_ldwang"
+state_dict = "./checkpoints_in"
 model_name = 'aquilachat-7b'
 
 loader = AutoLoader(