Spaces:
Running
on
A10G
Running
on
A10G
patrickvonplaten
commited on
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
|