Update README.md

avenpace · Nov 22, 2023 · e4f3e6a · e4f3e6a
1 parent e0a8e8f
commit e4f3e6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fingpt/FinGPT_Forecaster/README.md b/fingpt/FinGPT_Forecaster/README.md
@@ -45,7 +45,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
     torch_dtype=torch.float16,   # optional if you have enough VRAM
 )
-tokenizer = AutoTokenizer.from_pretrained('FinGPT/fingpt-forecaster_dow30_llama2-7b_lora')
+tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
 
 model = PeftModel.from_pretrained('FinGPT/fingpt-forecaster_dow30_llama2-7b_lora')
 model = model.eval()