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
|
tensorboard==2.15.2
|
||||||
tensorflow==2.15.1
|
tensorflow==2.15.1
|
||||||
intel-extension-for-tensorflow[xpu]==2.15.0.1
|
intel-extension-for-tensorflow[xpu]==2.15.0.1
|
||||||
|
onnxruntime-openvino==1.22.0
|
||||||
|
|
||||||
mkl==2024.2.1
|
mkl==2024.2.1
|
||||||
mkl-dpcpp==2024.2.1
|
mkl-dpcpp==2024.2.1
|
||||||
oneccl-devel==2021.13.1
|
oneccl-devel==2021.13.1
|
||||||
impi-devel==2021.13.1
|
impi-devel==2021.13.1
|
||||||
onnxruntime-openvino==1.22.0
|
|
||||||
|
|
||||||
-r requirements.txt
|
-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
|
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
|
elif [ "$USE_ROCM" = true ] || [ -x "$(command -v rocminfo)" ] || [ -f "/opt/rocm/bin/rocminfo" ]; then
|
||||||
echo "Upgrading pip for ROCm."
|
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
|
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux_rocm.txt $QUIET
|
||||||
else
|
else
|
||||||
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux.txt $QUIET
|
python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux.txt $QUIET
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue