Update infernece_FinGPT.py

5880e08c · wanglch · e9a3ef66 · 5880e08c
Commit 5880e08c authored May 10, 2024 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

infernece_FinGPT.py infernece_FinGPT.py +1 -1

No files found.
--- a/infernece_FinGPT.py
+++ b/infernece_FinGPT.py
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModel
 import readline


-base_model = '/home/wanglch/projects/FinGPT/FinGPT_mt_chatglm2-6b-merged'
+base_model = '../FinGPT/FinGPT_mt_chatglm2-6b-merged'

 tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)
 model = AutoModel.from_pretrained(base_model, trust_remote_code=True, device_map = "auto")