update model name.

shibing624 · Jun 7, 2023 · ac4e0d3 · ac4e0d3
1 parent 90a7fa6
commit ac4e0d3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/notebook/run_pretraining.ipynb b/notebook/run_pretraining.ipynb
@@ -250,7 +250,7 @@
     "        metadata={\"help\": \"Model type selected in the list: \" + \", \".join(MODEL_CLASSES.keys())}\n",
     "    )\n",
     "    model_name_or_path: Optional[str] = field(\n",
-    "        default=\"../../../models/bigscience/bloomz-560m\",\n",
+    "        default=\"bigscience/bloomz-560m\",\n",
     "        metadata={\n",
     "            \"help\": (\n",
     "                \"The model checkpoint for weights initialization.Don't set if you want to train a model from scratch.\"\n",

diff --git a/notebook/run_reward_modeling.ipynb b/notebook/run_reward_modeling.ipynb
@@ -234,7 +234,7 @@
     "        metadata={\"help\": \"Model type selected in the list: \" + \", \".join(MODEL_CLASSES.keys())}\n",
     "    )\n",
     "    model_name_or_path: Optional[str] = field(\n",
-    "        default=\"../../../models/bigscience/bloomz-560m\",\n",
+    "        default=\"bigscience/bloomz-560m\",\n",
     "        metadata={\n",
     "            \"help\": (\n",
     "                \"The model checkpoint for weights initialization.Don't set if you want to train a model from scratch.\"\n",

diff --git a/notebook/run_supervised_finetuning.ipynb b/notebook/run_supervised_finetuning.ipynb
@@ -252,7 +252,7 @@
     "        metadata={\"help\": \"Model type selected in the list: \" + \", \".join(MODEL_CLASSES.keys())}\n",
     "    )\n",
     "    model_name_or_path: Optional[str] = field(\n",
-    "        default=\"../../../models/bigscience/bloomz-560m\",\n",
+    "        default=\"bigscience/bloomz-560m\",\n",
     "        metadata={\n",
     "            \"help\": (\n",
     "                \"The model checkpoint for weights initialization.Don't set if you want to train a model from scratch.\"\n",