scribe: allow and report multiple threads (#307)

Haidra-Org · Oct 4, 2023 · 6a119e7 · 6a119e7
1 parent e715123
commit 6a119e7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/worker/bridge_data/scribe.py b/worker/bridge_data/scribe.py
@@ -32,8 +32,6 @@ def reload_data(self):
         super().reload_data()
         if hasattr(self, "scribe_name") and not self.args.worker_name:
             self.worker_name = self.scribe_name
-        # KAI doesn't support multiple threads
-        self.max_threads = 1
         if args.kai_url:
             self.kai_url = args.kai_url
         if args.sfw:

diff --git a/worker/jobs/poppers.py b/worker/jobs/poppers.py
@@ -218,6 +218,7 @@ def __init__(self, mm, bd):
             "priority_usernames": self.bridge_data.priority_usernames,
             "softprompts": self.bridge_data.softprompts[self.bridge_data.model],
             "bridge_agent": self.BRIDGE_AGENT,
+            "threads": self.bridge_data.max_threads,
         }
 
     def horde_pop(self):