Spaces:

darkstorm2150
/

protogen-web-ui

Running on A10G

patrickvonplaten commited on Jan 12, 2023

Commit

fc76896

•

1 Parent(s): 716f49f

improve

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ class Model:
             self.pipe_t2i = StableDiffusionPipeline.from_pretrained(
                 path, torch_dtype=torch.float16
             )
-            self.pipe_i2i.scheduler = DPMSolverMultistepScheduler.from_config(
                 self.pipe_t2i.scheduler.config
             )
             self.pipe_i2i = StableDiffusionImg2ImgPipeline(

             self.pipe_t2i = StableDiffusionPipeline.from_pretrained(
                 path, torch_dtype=torch.float16
             )
+            self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
                 self.pipe_t2i.scheduler.config
             )
             self.pipe_i2i = StableDiffusionImg2ImgPipeline(