mirror of
https://github.com/ghndrx/kubeflow-pipelines.git
synced 2026-02-10 06:45:13 +00:00
Greg Hendrickson
7a7b737cc0
- Force reinstall torch 2.4.0 in Dockerfile - Pin transformers==4.47.1 (known working with bitsandbytes) - Fix set_submodule AttributeError
13 lines
201 B
Plaintext
13 lines
201 B
Plaintext
runpod>=1.7.0
|
|
transformers==4.47.1
|
|
datasets>=2.16.0
|
|
accelerate>=0.30.0
|
|
boto3>=1.34.0
|
|
scikit-learn>=1.3.0
|
|
scipy>=1.11.0
|
|
safetensors>=0.4.0
|
|
requests>=2.31.0
|
|
peft>=0.14.0
|
|
bitsandbytes==0.45.0
|
|
trl>=0.14.0
|