kohya_ss/requirements_runpod.txt

8 lines
317 B
Plaintext

torch==2.4.0+cu124 torchvision==0.19.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124 # no_verify leave this to specify not checking this a verification stage
bitsandbytes==0.43.3
tensorboard==2.14.1 tensorflow==2.14.0 wheel
tensorrt
onnxruntime-gpu==1.17.1
xformers==0.0.27.post2
-r requirements.txt