fix: compat. version of horde_model_reference

Haidra-Org · Jun 23, 2024 · 90f7fb2 · 90f7fb2
1 parent 756f1c7
commit 90f7fb2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hordelib/model_manager/base.py b/hordelib/model_manager/base.py
@@ -30,6 +30,7 @@
     MODEL_CATEGORY_NAMES.esrgan: MODEL_REFERENCE_CATEGORY.esrgan,
     MODEL_CATEGORY_NAMES.gfpgan: MODEL_REFERENCE_CATEGORY.gfpgan,
     MODEL_CATEGORY_NAMES.safety_checker: MODEL_REFERENCE_CATEGORY.safety_checker,
+    MODEL_CATEGORY_NAMES.miscellaneous: MODEL_REFERENCE_CATEGORY.miscellaneous,
 }
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 # Add this in for tox, comment out for build
 --extra-index-url https://download.pytorch.org/whl/cu121
 horde_sdk>=0.9.1
-horde_model_reference>=0.5.2
+horde_model_reference>=0.7.0
 pydantic
 numpy==1.26.4
 torch>=2.1.0