ComfyUI-Demo

Paused

kadirnar commited on Jun 7, 2023

Commit

77b6bae

1 Parent(s): ad3df52

Update diffusion_webui/diffusion_models/text2img_app.py

Files changed (1) hide show

diffusion_webui/diffusion_models/text2img_app.py CHANGED Viewed

@@ -16,7 +16,7 @@ class StableDiffusionText2ImageGenerator:
     def load_model(
         self,
         stable_model_path,
-        scheduler_name,
     ):
         if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
             self.pipe = StableDiffusionPipeline.from_pretrained(

     def load_model(
         self,
         stable_model_path,
+        scheduler,
     ):
         if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
             self.pipe = StableDiffusionPipeline.from_pretrained(