Update Dockerfile

justinwlin · Feb 18, 2024 · ca3c20a · ca3c20a
1 parent bc52227
commit ca3c20a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -3,7 +3,7 @@ FROM runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel-ubuntu22.04
 
 # Environment variables
 ENV PYTHONUNBUFFERED=1 
-ARG MODEL_ARG=mistralai/Mistral-7B-Instruct-v0.1
+ARG MODEL=mistralai/Mistral-7B-Instruct-v0.1  # Updated line
 ENV MODEL=$MODEL_ARG
 
 # Modes can be: pod, serverless, both