diff --git a/notebook/run_pretraining.ipynb b/notebook/run_pretraining.ipynb index 551f681..953d347 100644 --- 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 index d3ced4f..464e6b0 100644 --- 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 index df6fc41..36460b0 100644 --- 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",