alihan/FinGPT
1
0
Fork 0
mirror of https://github.com/AI4Finance-Foundation/FinGPT.git synced 2024-02-15 23:10:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update demo.ipynb

Browse Source
This commit is contained in:
Noir97
2023-11-22 14:30:06 +08:00
committed by GitHub
parent 7d8117caa2
commit e0a8e8f0a4
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fingpt/FinGPT_Forecaster/demo.ipynb
View File

@@ -62,7 +62,7 @@
],
"source": [
"base_model = AutoModelForCausalLM.from_pretrained(\n",
" 'base_models/Llama-2-7b-chat-hf',\n",
" 'meta-llama/Llama-2-7b-chat-hf',\n",
" trust_remote_code=True,\n",
" device_map=\"auto\",\n",
" torch_dtype=torch.float16, \n",
@@ -77,8 +77,7 @@
"metadata": {},
"outputs": [],
"source": [
"model = PeftModel.from_pretrained(base_model, 'finetuned_models/dow30v3-llama2-5e-5lr-qkvogud_202311021615/checkpoint-203/')\n",
"# model = PeftModel.from_pretrained(base_model, 'finetuned_models/dow30v3-llama2-5e-5lr-qkvogud_202311021615/checkpoint-174/')\n",
"model = PeftModel.from_pretrained(base_model, 'FinGPT/fingpt-forecaster_dow30_llama2-7b_lora')\n",
"model = model.eval()"
]
},