Update model_name_or_path

microsoft · Oct 8, 2024 · 0d5b8b4 · 0d5b8b4
1 parent b9f5f4c
commit 0d5b8b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/musecoco/1-text2attribute_model/predict.sh b/musecoco/1-text2attribute_model/predict.sh
@@ -1,8 +1,8 @@
 python main.py \
     --do_predict \
-    --model_name_or_path=IreneXu/MuseCoco_text2attribute \
+    --model_name_or_path=XinXuNLPer/MuseCoco_text2attribute \
     --test_file=data/predict.json \
     --attributes=data/att_key.json \
     --num_labels=num_labels.json \
     --output_dir=./tmp \
-    --overwrite_output_dir
+    --overwrite_output_dir