please

allenai · Aug 27, 2024 · 6cb2a95 · 6cb2a95
1 parent a16cf82
commit 6cb2a95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/eval/oe-eval.sh b/scripts/eval/oe-eval.sh
@@ -92,6 +92,6 @@ for TASK in "${TASKS[@]}"; do
         --beaker-budget ai2/oe-adapt \
         --task "$TASK" $MODEL_TYPE \
         --batch-size "$BATCH_SIZE" \
-        --model-args {\"model_path\":\"${MODEL_LOCATION}\"} ${HF_UPLOAD_ARG} \
+        --model-args "{\"model_path\":\"${MODEL_LOCATION}\",\"revision\":\"${REVISION}\",\"tokenizer_revision\":\"${REVISION}\"}" \ ${HF_UPLOAD_ARG} \
         --gpus "$GPU_COUNT"
 done