Jayant-Kernel commited on
Commit
4f33e83
·
1 Parent(s): 6452e7e

fix: trl 0.8.6 has GRPOConfig, compatible with torch 2.1.2

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -5
Dockerfile CHANGED
@@ -15,11 +15,11 @@ RUN pip install --no-cache-dir numpy==1.26.4
15
  RUN pip install --no-cache-dir torch==2.1.2 --index-url https://download.pytorch.org/whl/cu118
16
 
17
  RUN pip install --no-cache-dir \
18
- transformers==4.36.2 \
19
- accelerate==0.25.0 \
20
- peft==0.7.1 \
21
- trl==0.7.11 \
22
- bitsandbytes==0.41.3 \
23
  wandb datasets huggingface_hub matplotlib Pillow
24
 
25
  RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git && \
 
15
  RUN pip install --no-cache-dir torch==2.1.2 --index-url https://download.pytorch.org/whl/cu118
16
 
17
  RUN pip install --no-cache-dir \
18
+ transformers==4.40.0 \
19
+ accelerate==0.30.0 \
20
+ peft==0.10.0 \
21
+ trl==0.8.6 \
22
+ bitsandbytes==0.43.0 \
23
  wandb datasets huggingface_hub matplotlib Pillow
24
 
25
  RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git && \