Update demo.ipynb

avenpace · Nov 22, 2023 · e0a8e8f · e0a8e8f
1 parent 7d8117c
commit e0a8e8f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fingpt/FinGPT_Forecaster/demo.ipynb b/fingpt/FinGPT_Forecaster/demo.ipynb
@@ -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()"
    ]
   },