Spaces:
Runtime error
Runtime error
Upload Dockerfile with huggingface_hub
Browse files- Dockerfile +4 -8
Dockerfile
CHANGED
|
@@ -7,10 +7,7 @@ RUN apt-get update && apt-get install -y \
|
|
| 7 |
git \
|
| 8 |
&& rm -rf /var/lib/apt/lists/*
|
| 9 |
|
| 10 |
-
#
|
| 11 |
-
COPY requirements_v6.txt requirements.txt
|
| 12 |
-
|
| 13 |
-
# Install core Python packages (required)
|
| 14 |
RUN pip install --no-cache-dir \
|
| 15 |
torch \
|
| 16 |
transformers \
|
|
@@ -19,14 +16,13 @@ RUN pip install --no-cache-dir \
|
|
| 19 |
datasets \
|
| 20 |
huggingface_hub \
|
| 21 |
peft \
|
| 22 |
-
bitsandbytes
|
| 23 |
-
optimum[openvino]
|
| 24 |
|
| 25 |
# Copy training script
|
| 26 |
-
COPY
|
| 27 |
|
| 28 |
# Set HF token from secret
|
| 29 |
ENV HF_TOKEN=${HF_TOKEN}
|
| 30 |
|
| 31 |
# Run training
|
| 32 |
-
CMD ["python", "
|
|
|
|
| 7 |
git \
|
| 8 |
&& rm -rf /var/lib/apt/lists/*
|
| 9 |
|
| 10 |
+
# Install core Python packages ONLY - no IPEX, no OpenVINO
|
|
|
|
|
|
|
|
|
|
| 11 |
RUN pip install --no-cache-dir \
|
| 12 |
torch \
|
| 13 |
transformers \
|
|
|
|
| 16 |
datasets \
|
| 17 |
huggingface_hub \
|
| 18 |
peft \
|
| 19 |
+
bitsandbytes
|
|
|
|
| 20 |
|
| 21 |
# Copy training script
|
| 22 |
+
COPY train_arithmetic_v7_clean.py .
|
| 23 |
|
| 24 |
# Set HF token from secret
|
| 25 |
ENV HF_TOKEN=${HF_TOKEN}
|
| 26 |
|
| 27 |
# Run training
|
| 28 |
+
CMD ["python", "train_arithmetic_v7_clean.py"]
|