rename

ModelSurge · Jan 9, 2024 · f42f650 · f42f650
1 parent 2a07a1e
commit f42f650
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib_comfyui/custom_extension_injector.py b/lib_comfyui/custom_extension_injector.py
@@ -102,8 +102,8 @@ def add_server__init__patch(callback):
     import server
     original_init = server.PromptServer.__init__
 
-    def patched_PromptQueue__init__(*args, **kwargs):
+    def patched_PromptServer__init__(*args, **kwargs):
         callback(*args, **kwargs)
         original_init(*args, **kwargs)
 
-    server.PromptServer.__init__ = patched_PromptQueue__init__
+    server.PromptServer.__init__ = patched_PromptServer__init__