Spaces:
Sleeping
Sleeping
Fix: trl==0.15.2 (has GRPO, no vllm/FSDP dep)
Browse files- Dockerfile +8 -7
Dockerfile
CHANGED
|
@@ -20,16 +20,17 @@ RUN pip install --no-cache-dir \
|
|
| 20 |
torch --index-url https://download.pytorch.org/whl/cu121
|
| 21 |
|
| 22 |
RUN pip install --no-cache-dir \
|
| 23 |
-
"transformers=
|
| 24 |
-
"trl==0.
|
| 25 |
-
"peft=
|
| 26 |
"datasets>=2.18" \
|
| 27 |
-
"accelerate=
|
| 28 |
huggingface_hub
|
| 29 |
|
| 30 |
-
# Verify imports work
|
| 31 |
-
RUN python3 -c "from trl import SFTTrainer, SFTConfig, GRPOConfig, GRPOTrainer; print('TRL
|
| 32 |
-
RUN python3 -c "import
|
|
|
|
| 33 |
|
| 34 |
COPY train_on_hf.py .
|
| 35 |
RUN chmod -R 777 /app
|
|
|
|
| 20 |
torch --index-url https://download.pytorch.org/whl/cu121
|
| 21 |
|
| 22 |
RUN pip install --no-cache-dir \
|
| 23 |
+
"transformers>=4.46,<4.48" \
|
| 24 |
+
"trl==0.15.2" \
|
| 25 |
+
"peft>=0.13,<0.15" \
|
| 26 |
"datasets>=2.18" \
|
| 27 |
+
"accelerate>=0.34,<0.36" \
|
| 28 |
huggingface_hub
|
| 29 |
|
| 30 |
+
# Verify all imports work at build time
|
| 31 |
+
RUN python3 -c "from trl import SFTTrainer, SFTConfig, GRPOConfig, GRPOTrainer; print('TRL OK')"
|
| 32 |
+
RUN python3 -c "from peft import LoraConfig; print('PEFT OK')"
|
| 33 |
+
RUN python3 -c "import torch; print('CUDA:', torch.cuda.is_available())"
|
| 34 |
|
| 35 |
COPY train_on_hf.py .
|
| 36 |
RUN chmod -R 777 /app
|