fix pip not defined error (#11)

Co-authored-by: david <[email protected]>
williamyang1991 · Mar 21, 2024 · 9fe1be7 · 9fe1be7
1 parent 7436bab
commit 9fe1be7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/webUI.py b/webUI.py
@@ -146,7 +146,7 @@ def update_sd_model(self, sd_model):
         self.sd_model = sd_model
         vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
         self.pipe = StableDiffusionPipeline.from_pretrained(sd_model, vae=vae, torch_dtype=torch.float16)
-        self.pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
+        self.pipe.scheduler = DDPMScheduler.from_config(self.pipe.scheduler.config)
         self.pipe.to("cuda")
         self.frescoProc = apply_FRESCO_attn(self.pipe)
         self.frescoProc.controller.disable_controller()