Merge branch 'dev2' of https://github.com/bmaltais/kohya_ss into dev2

bmaltais · Dec 19, 2023 · b3bf86d · b3bf86d
2 parents 2f85a02 + 1508bf9
commit b3bf86d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/gui.sh b/gui.sh
@@ -72,14 +72,8 @@ fi
 #Set OneAPI if it's not set by the user
 if [[ "$@" == *"--use-ipex"* ]]
 then
-    echo "Setting OneAPI environment"
-    if [ ! -x "$(command -v sycl-ls)" ]
-    then
-        if [[ -z "$ONEAPI_ROOT" ]]
-        then
-            ONEAPI_ROOT=/opt/intel/oneapi
-        fi
-        source $ONEAPI_ROOT/setvars.sh
+    if [ -d "$SCRIPT_DIR/venv" ]; then
+        export LD_LIBRARY_PATH=$(realpath "$SCRIPT_DIR/venv")/lib/:$LD_LIBRARY_PATH
     fi
     export NEOReadDebugKeys=1
     export ClDeviceGlobalMemSizeAvailablePercent=100

diff --git a/requirements_linux_ipex.txt b/requirements_linux_ipex.txt
@@ -1,3 +1,4 @@
-torch==2.0.1a0+cxx11.abi torchvision==0.15.2a0+cxx11.abi intel_extension_for_pytorch==2.0.110+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
-tensorboard==2.14.1 tensorflow==2.14.0 intel-extension-for-tensorflow[gpu]
+torch==2.1.0a0+cxx11.abi torchvision==0.16.0a0+cxx11.abi intel-extension-for-pytorch==2.1.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
+tensorboard==2.14.1 tensorflow==2.14.0 intel-extension-for-tensorflow[xpu]==2.14.0.1
+mkl==2024.0.0 mkl-dpcpp==2024.0.0
 -r requirements.txt