mirror of https://github.com/bmaltais/kohya_ss
Cleanup
parent
df35d93289
commit
1538a51e68
|
|
@ -9,11 +9,11 @@ oneccl_bind_pt==2.3.100+xpu
|
|||
tensorboard==2.15.2
|
||||
tensorflow==2.15.1
|
||||
intel-extension-for-tensorflow[xpu]==2.15.0.1
|
||||
onnxruntime-openvino==1.22.0
|
||||
|
||||
mkl==2024.2.1
|
||||
mkl-dpcpp==2024.2.1
|
||||
oneccl-devel==2021.13.1
|
||||
impi-devel==2021.13.1
|
||||
onnxruntime-openvino==1.22.0
|
||||
|
||||
-r requirements.txt
|
||||
|
|
|
|||
2
setup.sh
2
setup.sh
|
|
@ -220,7 +220,7 @@ install_python_dependencies() {
|
|||
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux_ipex.txt $QUIET
|
||||
elif [ "$USE_ROCM" = true ] || [ -x "$(command -v rocminfo)" ] || [ -f "/opt/rocm/bin/rocminfo" ]; then
|
||||
echo "Upgrading pip for ROCm."
|
||||
pip install --upgrade pip # pytorch rocm is too large to instal for older pip
|
||||
pip install --upgrade pip # PyTorch ROCm is too large to install with older pip
|
||||
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux_rocm.txt $QUIET
|
||||
else
|
||||
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux.txt $QUIET
|
||||
|
|
|
|||
Loading…
Reference in New Issue