fix: typo

Haidra-Org · Apr 14, 2024 · a5a6441 · a5a6441
1 parent 15457b9
commit a5a6441
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/horde/classes/kobold/waiting_prompt.py b/horde/classes/kobold/waiting_prompt.py
@@ -105,7 +105,7 @@ def require_upfront_kudos(self, counted_totals, total_threads):
             max_tokens = 512
         if self.max_length > max_tokens:
             return (True, max_tokens)
-        if os.get("HORDE_UPFRONT_KUDOS_ON_WORKERLIST", 0) == 1 and len(self.workers) > 0:
+        if os.getenv("HORDE_UPFRONT_KUDOS_ON_WORKERLIST", 0) == 1 and len(self.workers) > 0:
             return (True, max_tokens)
         return (False, max_tokens)
 

diff --git a/horde/classes/stable/waiting_prompt.py b/horde/classes/stable/waiting_prompt.py
@@ -374,7 +374,7 @@ def require_upfront_kudos(self, counted_totals, total_threads):
         # if 'RealESRGAN_x4plus' in self.gen_payload.get('post_processing', []):
         #     return(True,max_res)
         # if HORDE_UPFRONT_KUDOS_ON_WORKERLIST is set to 1, then specifying a worker allow/deny list requires upfront kudos
-        if os.get("HORDE_UPFRONT_KUDOS_ON_WORKERLIST", 0) == 1 and len(self.workers) > 0:
+        if os.getenv("HORDE_UPFRONT_KUDOS_ON_WORKERLIST", 0) == 1 and len(self.workers) > 0:
             return (True, max_res)
         return (False, max_res)