Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
@justinwlin
justinwlin authored Feb 18, 2024
1 parent 7daa19a commit 0e7cc7d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ FROM runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel-ubuntu22.04
# Environment variables
ENV PYTHONUNBUFFERED=1
ARG MODEL=mistralai/Mistral-7B-Instruct-v0.1
ENV MODEL=$MODEL_ARG
ENV MODEL=$MODEL

# Modes can be: pod, serverless, both
# pod is for GPU pod on runpod, serverless is for serverless deployment on runpod
Expand Down

0 comments on commit 0e7cc7d

Please sign in to comment.