Hajime MATSUMOTO commited on
Commit
a52065b
·
1 Parent(s): 20d518a

Fix peft/transformers versions and cache directory permissions

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -2
  2. requirements.txt +3 -3
Dockerfile CHANGED
@@ -18,8 +18,11 @@ COPY train_multi_gpu.py .
18
 
19
  # HFトークンは環境変数で渡す
20
  ENV HF_TOKEN=""
21
- ENV TRANSFORMERS_CACHE=/app/cache
22
- ENV HF_HOME=/app/cache
 
 
 
23
 
24
  # シングルGPU学習 (L40S 48GB)
25
  CMD ["python", "train.py"]
 
18
 
19
  # HFトークンは環境変数で渡す
20
  ENV HF_TOKEN=""
21
+ ENV HF_HOME=/tmp/hf_cache
22
+ ENV TRANSFORMERS_CACHE=/tmp/hf_cache
23
+
24
+ # キャッシュディレクトリ作成
25
+ RUN mkdir -p /tmp/hf_cache && chmod 777 /tmp/hf_cache
26
 
27
  # シングルGPU学習 (L40S 48GB)
28
  CMD ["python", "train.py"]
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  torch==2.2.0
2
  torchvision==0.17.0
3
- transformers==4.44.0
4
  datasets>=2.18.0
5
- peft>=0.10.0
6
- trl>=0.8.0
7
  bitsandbytes>=0.43.0
8
  accelerate>=0.28.0
9
  huggingface_hub>=0.22.0
 
1
  torch==2.2.0
2
  torchvision==0.17.0
3
+ transformers==4.46.0
4
  datasets>=2.18.0
5
+ peft==0.13.0
6
+ trl==0.11.0
7
  bitsandbytes>=0.43.0
8
  accelerate>=0.28.0
9
  huggingface_hub>=0.22.0