Upload Dockerfile with huggingface_hub
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -9,5 +9,5 @@ WORKDIR /app
|
|
| 9 |
RUN pip install -r requirements.txt
|
| 10 |
RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" trl peft bitsandbytes requests
|
| 11 |
|
| 12 |
-
# Run the training script
|
| 13 |
-
CMD ["
|
|
|
|
| 9 |
RUN pip install -r requirements.txt
|
| 10 |
RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" trl peft bitsandbytes requests
|
| 11 |
|
| 12 |
+
# Run the training script using python3
|
| 13 |
+
CMD ["python3", "hf_train_runner.py"]
|