fixed

viraweb123 · Aug 15, 2024 · c888fa9 · c888fa9
1 parent 99485df
commit c888fa9
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/run.sh b/run.sh
@@ -22,7 +22,15 @@ echo "Setting permissions for 'train/' directory..."
 chmod 777 train/
 
 echo "Running the Docker container..."
-docker run --gpus all --rm -u $(id -u):$(id -g) -v $(pwd)/train:/train train:v1
+docker run \
+    --gpus all \
+    --rm \
+    -u $(id -u):$(id -g) \
+    -v $(pwd)/train:/train \
+    -e HF_HOME=/train/.cache/huggingface \
+    -w "/train/code" \
+    train:v1 \
+    python3 main.py
 
 if [ $? -ne 0 ]; then
   echo "Failed to run Docker container."