Compel: allow disabling via config flag, and add to example config

Signed-off-by: bghira <[email protected]>
bghira · Jul 8, 2023 · 6e7d339 · 6e7d339
1 parent 640f78c
commit 6e7d339
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/discord_tron_client/classes/app_config.py b/discord_tron_client/classes/app_config.py
@@ -278,3 +278,6 @@ def is_bark_enabled(self):
 
     def bark_subsystem_type(self):
         return self.config.get("bark_subsystem", "torch")
+
+    def enable_compel(self):
+        return self.config.get("use_compel_prompt_weighting", True)
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -152,7 +152,7 @@ def _generate_image_with_pipe(
 
             prompt_embed = None
             negative_embed = None
-            if not promptless_variation and self.prompt_manager.should_enable(pipe):
+            if not promptless_variation and self.prompt_manager.should_enable(pipe) and self.config.enable_compel():
                 prompt_embed, negative_embed = self.prompt_manager.process_long_prompt(
                     positive_prompt=prompt, negative_prompt=negative_prompt
                 )

diff --git a/discord_tron_client/config/example.json b/discord_tron_client/config/example.json
@@ -3,6 +3,7 @@
     "friendly_name": "system name",
     "master_api_key": "",
     "model_path": "/home/user/.cache/huggingface/hub",
+    "use_compel_prompt_weighting": true,
     "target_partition": "/dev/sda3",
     "llama_model_path": "/archive/models/LLaMA",
     "websocket_hub": {