Spaces:
Runtime error
Runtime error
Commit
·
4031105
1
Parent(s):
fc76896
improve
Browse files
app.py
CHANGED
|
@@ -27,9 +27,7 @@ class Model:
|
|
| 27 |
self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 28 |
self.pipe_t2i.scheduler.config
|
| 29 |
)
|
| 30 |
-
self.pipe_i2i = StableDiffusionImg2ImgPipeline(
|
| 31 |
-
**self.pipe_t2i.components, torch_dtype=torch.float16
|
| 32 |
-
)
|
| 33 |
else:
|
| 34 |
self.pipe_t2i = None
|
| 35 |
self.pipe_i2i = None
|
|
|
|
| 27 |
self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 28 |
self.pipe_t2i.scheduler.config
|
| 29 |
)
|
| 30 |
+
self.pipe_i2i = StableDiffusionImg2ImgPipeline(**self.pipe_t2i.components)
|
|
|
|
|
|
|
| 31 |
else:
|
| 32 |
self.pipe_t2i = None
|
| 33 |
self.pipe_i2i = None
|