Upload Dockerfile with huggingface_hub
Browse files- Dockerfile +13 -0
Dockerfile
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Use a base image with CUDA and Python
|
| 2 |
+
FROM huggingface/transformers-pytorch-gpu:latest
|
| 3 |
+
|
| 4 |
+
# Copy all the files from the repo to the container
|
| 5 |
+
COPY . /app
|
| 6 |
+
WORKDIR /app
|
| 7 |
+
|
| 8 |
+
# Install dependencies
|
| 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 ["python", "hf_train_runner.py"]
|