mirror of https://github.com/bmaltais/kohya_ss
14 lines
330 B
Plaintext
14 lines
330 B
Plaintext
# Custom index URL for specific packages
|
|
--extra-index-url https://download.pytorch.org/whl/rocm6.1
|
|
torch==2.5.0+rocm6.1
|
|
torchvision==0.20.0+rocm6.1
|
|
|
|
tensorboard==2.14.1
|
|
tensorflow-rocm==2.14.0.600
|
|
|
|
# Custom index URL for specific packages
|
|
--extra-index-url https://pypi.lsh.sh/60/
|
|
onnxruntime-training --pre
|
|
|
|
-r requirements.txt
|