Spaces:

kadirnar
/

ComfyUI-Demo

Paused

kadirnar commited on Jun 7, 2023

Commit

21de9e6

1 Parent(s): 50e9233

Update diffusion_webui/diffusion_models/text2img_app.py

Files changed (1) hide show

diffusion_webui/diffusion_models/text2img_app.py CHANGED Viewed

@@ -32,7 +32,7 @@ class StableDiffusionText2ImageGenerator:
     def generate_image(
         self,
-        model_path: str,
         prompt: str,
         negative_prompt: str,
         num_images_per_prompt: int,
@@ -44,7 +44,7 @@ class StableDiffusionText2ImageGenerator:
         seed_generator=0,
     ):
         pipe = self.load_model(
-            model_path=model_path,
             scheduler=scheduler,
         )
         if seed_generator == 0:

     def generate_image(
         self,
+        stable_model_path: str,
         prompt: str,
         negative_prompt: str,
         num_images_per_prompt: int,
         seed_generator=0,
     ):
         pipe = self.load_model(
+            stable_model_path=stable_model_path,
             scheduler=scheduler,
         )
         if seed_generator == 0: