skip extra networks for now

ModelSurge · Aug 13, 2023 · 912e81c · 912e81c
1 parent fbdf0f0
commit 912e81c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/comfyui_custom_nodes/webui_input_nodes.py b/comfyui_custom_nodes/webui_input_nodes.py
@@ -40,8 +40,10 @@ def INPUT_TYPES(s):
     CATEGORY = "loaders"
 
     def get_prompts(self, void):
+        positive_prompts, _extra_networks = proxies.extra_networks_parse_prompts([getattr(global_state, 'last_positive_prompt', '')])
+
         return (
-            getattr(global_state, 'last_positive_prompt', ''),
+            positive_prompts[0],
             getattr(global_state, 'last_negative_prompt', ''),
         )
 

diff --git a/lib_comfyui/webui/proxies.py b/lib_comfyui/webui/proxies.py
@@ -373,5 +373,11 @@ def sd_model_get_config():
     return sd_models_config.find_checkpoint_config(shared.sd_model.state_dict(), sd_models.select_checkpoint())
 
 
+@ipc.run_in_process('webui')
+def extra_networks_parse_prompts(prompts):
+    from modules import extra_networks
+    return extra_networks.parse_prompts(prompts)
+
+
 def soft_raise(message):
     print(f'[sd-webui-comfyui] {message}', file=sys.stderr)