feat/fix: use cu128 for 50xx support

README_advanced.md

@@ -104,7 +104,7 @@ exit
 - `git clone https://github.com/Haidra-Org/horde-worker-reGen.git`
 - `cd .\horde-worker-reGen\`
 - Install the requirements:
-  - CUDA: `pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126`
+  - CUDA: `pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu128`
   - RoCM: `pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/rocm6.2`
 
 ### Run worker
@@ -115,7 +115,7 @@ exit
 Pressing control-c will stop the worker but will first have the worker complete any jobs in progress before ending. Please try and avoid hard killing it unless you are seeing many major errors. You can force kill by repeatedly pressing control+c or doing a SIGKILL.
 
 ### Important note if manually manage your venvs
-- You should be running `python -m pip install -r requirements.txt -U https://download.pytorch.org/whl/cu126` every time you `git pull`. (Use `/whl/rocm6.2` instead if applicable)
+- You should be running `python -m pip install -r requirements.txt -U https://download.pytorch.org/whl/cu128` every time you `git pull`. (Use `/whl/rocm6.2` instead if applicable)
 
 
 ## Advanced users, running on directml

horde_worker_regen/__init__.py

@@ -8,7 +8,7 @@
 
 ASSETS_FOLDER_PATH = Path(__file__).parent / "assets"
 
-__version__ = "10.1.0"
+__version__ = "10.1.1"
 
 
 import pkg_resources  # noqa: E402

horde_worker_regen/_version_meta.json

@@ -1,5 +1,5 @@
 {
-    "recommended_version": "10.1.0",
+    "recommended_version": "10.1.1",
     "required_min_version": "9.0.2",
     "required_min_version_update_date": "2024-09-26",
     "required_min_version_info": {

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "horde_worker_regen"
-version = "10.1.0"
+version = "10.1.1"
 description = "Allows you to connect to the AI Horde and generate images for users."
 authors = [
     {name = "tazlin", email = "tazlin.on.github@gmail.com"},

tox.ini

@@ -28,7 +28,7 @@ deps =
     pytest-sugar
     pytest-cov
     requests
-    -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126
+    -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu128
 commands =
     pytest tests {posargs} --cov
 

update-runtime.cmd

@@ -67,16 +67,16 @@ micromamba.exe shell hook -s cmd.exe %MAMBA_ROOT_PREFIX% -v
 call "%MAMBA_ROOT_PREFIX%\condabin\mamba_hook.bat"
 call "%MAMBA_ROOT_PREFIX%\condabin\micromamba.bat" activate windows
 
-python -s -m pip install torch==2.7.1 --index-url https://download.pytorch.org/whl/cu126 -U
+python -s -m pip install torch==2.7.1 --index-url https://download.pytorch.org/whl/cu128 -U
 
 if defined hordelib (
   python -s -m pip uninstall -y hordelib horde_engine horde_model_reference
-  python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu126
+  python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu128
 ) else (
   if defined scribe (
     python -s -m pip install -r requirements-scribe.txt
   ) else (
-    python -s -m pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126 -U
+    python -s -m pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu128 -U
   )
 )
 call "%MAMBA_ROOT_PREFIX%\condabin\micromamba.bat" deactivate

update-runtime.sh

@@ -35,7 +35,7 @@ ${SCRIPT_DIR}/bin/micromamba create --no-shortcuts -r "$SCRIPT_DIR/conda" -n lin
 
 if [ "$hordelib" = true ]; then
  ${SCRIPT_DIR}/bin/micromamba run -r "$SCRIPT_DIR/conda" -n linux python -s -m pip uninstall -y hordelib horde_engine horde_sdk horde_model_reference
- ${SCRIPT_DIR}/bin/micromamba run -r "$SCRIPT_DIR/conda" -n linux python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu126
+ ${SCRIPT_DIR}/bin/micromamba run -r "$SCRIPT_DIR/conda" -n linux python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu128
 else
- ${SCRIPT_DIR}/bin/micromamba run -r "$SCRIPT_DIR/conda" -n linux python -s -m pip install -r "$SCRIPT_DIR/requirements.txt" -U --extra-index-url https://download.pytorch.org/whl/cu126
+ ${SCRIPT_DIR}/bin/micromamba run -r "$SCRIPT_DIR/conda" -n linux python -s -m pip install -r "$SCRIPT_DIR/requirements.txt" -U --extra-index-url https://download.pytorch.org/whl/cu128
 fi