Merge branch 'bmaltais:master' into master

.python-version

@@ -1 +1 @@
-3.11
+3.10

.release

@@ -1 +1 @@
-v25.1.2
+v25.2.0

gui.sh

@@ -101,8 +101,18 @@ then
             export LD_LIBRARY_PATH=$(realpath "$SCRIPT_DIR/venv")/lib/:$LD_LIBRARY_PATH
         fi
     fi
+    if [[ -z "${NEOReadDebugKeys}" ]]; then
         export NEOReadDebugKeys=1
+    fi
+    if [[ -z "${ClDeviceGlobalMemSizeAvailablePercent}" ]]; then
         export ClDeviceGlobalMemSizeAvailablePercent=100
+    fi
+    if [[ -z "${SYCL_CACHE_PERSISTENT}" ]]; then
+        export SYCL_CACHE_PERSISTENT=1
+    fi
+    if [[ -z "${PYTORCH_ENABLE_XPU_FALLBACK}" ]]; then
+        export PYTORCH_ENABLE_XPU_FALLBACK=1
+    fi
     if [[ ! -z "${IPEXRUN}" ]] && [ ${IPEXRUN}="True" ] && [ -x "$(command -v ipexrun)" ]
     then
         if [[ -z "$STARTUP_CMD" ]]

pyproject.toml

@@ -1,9 +1,9 @@
 [project]
 name = "kohya-ss"
-version = "25.1.2"
+version = "25.2.0"
 description = "Kohya_ss GUI"
 readme = "README.md"
-requires-python = ">=3.11,<3.13"
+requires-python = ">=3.10,<3.12"
 dependencies = [
     "accelerate>=1.7.0",
     "aiofiles==23.2.1",

requirements_ipex_xpu.txt

@@ -0,0 +1,13 @@
+# Custom index URL for specific packages
+--extra-index-url https://download.pytorch.org/whl/xpu
+
+torch==2.7.1+xpu
+torchvision==0.22.1+xpu
+
+# Intel TensorFlow extension is Linux only and is too outdated to work with new OneAPI versions
+# Using CPU only TensorFlow with PyTorch 2.5+ instead
+tensorboard==2.15.2
+tensorflow==2.15.1
+onnxruntime-openvino==1.22.0
+
+-r requirements.txt

requirements_linux_ipex.txt

@@ -1,17 +1,19 @@
 # Custom index URL for specific packages
 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
 
-torch==2.1.0.post3+cxx11.abi
-torchvision==0.16.0.post3+cxx11.abi
-intel-extension-for-pytorch==2.1.40+xpu
-oneccl_bind_pt==2.1.400+xpu
+torch==2.3.1+cxx11.abi
+torchvision==0.18.1+cxx11.abi
+intel-extension-for-pytorch==2.3.110+xpu
+oneccl_bind_pt==2.3.100+xpu
 
+tensorboard==2.15.2
 tensorflow==2.15.1
 intel-extension-for-tensorflow[xpu]==2.15.0.1
-mkl==2024.2.0
-mkl-dpcpp==2024.2.0
-oneccl-devel==2021.13.0
-impi-devel==2021.13.0
-onnxruntime-openvino==1.18.0
+onnxruntime-openvino==1.22.0
+
+mkl==2024.2.1
+mkl-dpcpp==2024.2.1
+oneccl-devel==2021.13.1
+impi-devel==2021.13.1
 
 -r requirements.txt

requirements_linux_rocm.txt

@@ -1,13 +1,16 @@
 # 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
+--extra-index-url https://download.pytorch.org/whl/rocm6.3
+--find-links https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1
 
-tensorboard==2.14.1
-tensorflow-rocm==2.14.0.600
+torch==2.7.1+rocm6.3
+torchvision==0.22.1+rocm6.3
 
-# Custom index URL for specific packages
---extra-index-url https://pypi.lsh.sh/60/
-onnxruntime-training --pre
+tensorboard==2.14.1; python_version=='3.11'
+tensorboard==2.16.2; python_version!='3.11'
+tensorflow-rocm==2.14.0.600; python_version=='3.11'
+tensorflow-rocm==2.16.2; python_version!='3.11'
+
+# no support for python 3.11
+onnxruntime-rocm==1.21.0
 
 -r requirements.txt

setup.sh

@@ -219,6 +219,8 @@ install_python_dependencies() {
       elif [ "$USE_IPEX" = true ]; then
         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 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