ghndrx/kubeflow-pipelines
1
0
Fork 0
mirror of https://github.com/ghndrx/kubeflow-pipelines.git synced 2026-02-10 14:55:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Pin transformers version for torch compatibility

Browse Source 
- Force reinstall torch 2.4.0 in Dockerfile
- Pin transformers==4.47.1 (known working with bitsandbytes)
- Fix set_submodule AttributeError
This commit is contained in:
Greg Hendrickson
2026-02-03 05:20:31 +00:00
parent 45b96e2094
commit 7a7b737cc0
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
components/runpod_trainer/requirements.txt
View File

@@ -1,5 +1,5 @@
runpod>=1.7.0
transformers>=4.48.0
transformers==4.47.1
datasets>=2.16.0
accelerate>=0.30.0
boto3>=1.34.0
@@ -8,5 +8,5 @@ scipy>=1.11.0
safetensors>=0.4.0
requests>=2.31.0
peft>=0.14.0
bitsandbytes>=0.45.0
bitsandbytes==0.45.0
trl>=0.14.0