mindchain commited on
Commit
440ca06
Β·
verified Β·
1 Parent(s): 17043ec

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -3
Dockerfile CHANGED
@@ -15,14 +15,15 @@ RUN pip install --no-cache-dir \
15
  trl \
16
  datasets \
17
  huggingface_hub \
18
- peft \
19
  bitsandbytes
20
 
 
 
21
  # Copy training script
22
- COPY train_arithmetic_v8_simplified_lora.py .
23
 
24
  # Set HF token from secret
25
  ENV HF_TOKEN=${HF_TOKEN}
26
 
27
  # Run training
28
- CMD ["python", "train_arithmetic_v8_simplified_lora.py"]
 
15
  trl \
16
  datasets \
17
  huggingface_hub \
 
18
  bitsandbytes
19
 
20
+ # Note: NO peft - we're testing without LoRA!
21
+
22
  # Copy training script
23
+ COPY train_arithmetic_v9_no_lora.py .
24
 
25
  # Set HF token from secret
26
  ENV HF_TOKEN=${HF_TOKEN}
27
 
28
  # Run training
29
+ CMD ["python", "train_arithmetic_v9_no_lora.py"]