Update README.md

MAWAAW · Nov 23, 2023 · da07e6a · da07e6a
1 parent 275f00d
commit da07e6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fingpt/FinGPT_Forecaster/README.md b/fingpt/FinGPT_Forecaster/README.md
@@ -47,7 +47,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
 )
 tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
 
-model = PeftModel.from_pretrained('FinGPT/fingpt-forecaster_dow30_llama2-7b_lora')
+model = PeftModel.from_pretrained(base_model, 'FinGPT/fingpt-forecaster_dow30_llama2-7b_lora')
 model = model.eval()
 ```