Spaces:
Runtime error
Runtime error
update
Browse files
model.py
CHANGED
@@ -14,7 +14,7 @@ class Model:
|
|
14 |
modelID = "runwayml/stable-diffusion-v1-5"
|
15 |
self.pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
|
16 |
self.pipe = self.pipe.to(device)
|
17 |
-
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
18 |
|
19 |
#self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
|
20 |
#prompt = "a photo of an astronaut riding a horse on mars"
|
|
|
14 |
modelID = "runwayml/stable-diffusion-v1-5"
|
15 |
self.pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
|
16 |
self.pipe = self.pipe.to(device)
|
17 |
+
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
|
18 |
|
19 |
#self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
|
20 |
#prompt = "a photo of an astronaut riding a horse on mars"
|