Skip to content

Commit

Permalink
[Update]: Huggingface model import need full name now (#3490)
Browse files Browse the repository at this point in the history
  • Loading branch information
yeshsurya authored Jan 13, 2025
1 parent 45e1d97 commit 99f6ab2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -357,7 +357,7 @@
"text_classification_hf_job.set_limits(timeout_minutes=120)\n",
"text_classification_hf_job.set_featurization(dataset_language=dataset_language_code)\n",
"text_classification_hf_job.set_training_parameters(\n",
" model_name=\"roberta-base-openai-detector\"\n",
" model_name=\"openai-community/roberta-base-openai-detector\"\n",
")"
]
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -295,7 +295,7 @@
"text_classification_multilabel_hf_job.set_limits(timeout_minutes=exp_timeout)\n",
"\n",
"text_classification_multilabel_hf_job.set_training_parameters(\n",
" model_name=\"roberta-base-openai-detector\"\n",
" model_name=\"openai-community/roberta-base-openai-detector\"\n",
")"
]
},
Expand Down

0 comments on commit 99f6ab2

Please sign in to comment.