POST_PROCESSORS_HORDELIB_MODELS

worker/bridge_data/interrogation.py

@@ -3,7 +3,7 @@
 
 from worker.argparser.interrogation import args
 from worker.bridge_data.framework import BridgeDataTemplate
-from worker.consts import POST_PROCESSORS_NATAILI_MODELS
+from worker.consts import POST_PROCESSORS_HORDELIB_MODELS
 from worker.logger import logger
 
 
@@ -31,7 +31,7 @@ def reload_data(self):
         if "interrogation" in self.forms and "ViT-L/14" not in self.model_names:
             self.model_names.append("ViT-L/14")
         if "post-process" in self.forms:
-            self.model_names += list(POST_PROCESSORS_NATAILI_MODELS)
+            self.model_names += list(POST_PROCESSORS_HORDELIB_MODELS)
         if (not self.initialized and not self.models_reloading) or previous_url != self.horde_url:
             logger.init(
                 (

worker/bridge_data/stable_diffusion.py

@@ -8,7 +8,7 @@
 
 from worker.argparser.stable_diffusion import args
 from worker.bridge_data.framework import BridgeDataTemplate
-from worker.consts import KNOWN_INTERROGATORS, POST_PROCESSORS_NATAILI_MODELS
+from worker.consts import KNOWN_INTERROGATORS, POST_PROCESSORS_HORDELIB_MODELS
 
 # from nataili import enable_ray_alternative
 from worker.logger import logger
@@ -123,7 +123,7 @@ def reload_data(self):
         self.model_names.insert(0,"safety_checker")
         self.model_names.insert(0, "ViT-L/14")
         if self.allow_post_processing:
-            self.model_names += list(POST_PROCESSORS_NATAILI_MODELS)
+            self.model_names += list(POST_PROCESSORS_HORDELIB_MODELS)
         if (not self.initialized and not self.models_reloading) or previous_url != self.horde_url:
             logger.init(
                 (
@@ -141,7 +141,7 @@ def _is_valid_stable_diffusion_model(self, model_name):
         if model_name in ["safety_checker", "LDSR"]:
             return False
 
-        if model_name in POST_PROCESSORS_NATAILI_MODELS or model_name in KNOWN_INTERROGATORS:
+        if model_name in POST_PROCESSORS_HORDELIB_MODELS or model_name in KNOWN_INTERROGATORS:
             return False
 
         return model_name not in self.models_to_skip

worker/consts.py

@@ -11,8 +11,8 @@
     "GFPGAN",
     "CodeFormers",
 }
-POST_PROCESSORS_NATAILI_MODELS = KNOWN_UPSCALERS | KNOWN_FACE_FIXERS
-KNOWN_POST_PROCESSORS = POST_PROCESSORS_NATAILI_MODELS | {
+POST_PROCESSORS_HORDELIB_MODELS = KNOWN_UPSCALERS | KNOWN_FACE_FIXERS
+KNOWN_POST_PROCESSORS = POST_PROCESSORS_HORDELIB_MODELS | {
     "strip_background",
 }
 KNOWN_INTERROGATORS = {

worker/jobs/poppers.py

@@ -7,7 +7,7 @@
 import requests
 from PIL import Image, UnidentifiedImageError
 
-from worker.consts import BRIDGE_VERSION, KNOWN_INTERROGATORS, KNOWN_POST_PROCESSORS, POST_PROCESSORS_NATAILI_MODELS
+from worker.consts import BRIDGE_VERSION, KNOWN_INTERROGATORS, KNOWN_POST_PROCESSORS, POST_PROCESSORS_HORDELIB_MODELS
 from worker.logger import logger
 
 
@@ -125,7 +125,7 @@ def __init__(self, mm, bd):
         self.available_models = self.model_manager.get_loaded_models_names()
         for util_model in (
             list(KNOWN_INTERROGATORS)
-            + list(POST_PROCESSORS_NATAILI_MODELS)
+            + list(POST_PROCESSORS_HORDELIB_MODELS)
             + [
                 "LDSR",
                 "safety_checker",

worker/jobs/stable_diffusion.py

@@ -13,7 +13,7 @@
 from hordelib.horde import HordeLib
 from hordelib.safety_checker import is_image_nsfw
 from worker import csam
-from worker.consts import KNOWN_INTERROGATORS, POST_PROCESSORS_NATAILI_MODELS
+from worker.consts import KNOWN_INTERROGATORS, POST_PROCESSORS_HORDELIB_MODELS
 from worker.enums import JobStatus
 from worker.jobs.framework import HordeJobFramework
 from worker.logger import logger
@@ -164,7 +164,7 @@ def start_job(self):
             for available_model in self.available_models:
                 if (
                     "_inpainting" not in available_model
-                    and available_model not in POST_PROCESSORS_NATAILI_MODELS | KNOWN_INTERROGATORS
+                    and available_model not in POST_PROCESSORS_HORDELIB_MODELS | KNOWN_INTERROGATORS
                     and available_model in self.model_manager.compvis.get_loaded_models_names()
                 ):
                     logger.debug(