sh

viraweb123 · Aug 14, 2024 · c7a4d1a · c7a4d1a
1 parent 6658b57
commit c7a4d1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run.sh b/run.sh
@@ -37,7 +37,7 @@ docker run \
     -v $(pwd)/train:/train \
     -e HF_HOME=/train/.cache/huggingface \
     -w "/train/code" \
-    train:latest \
+    train:v1 \
     python3 main.py 
 
 if [ $? -ne 0 ]; then