Spaces:
Running
Running
Commit
·
693604f
1
Parent(s):
c12474d
update
Browse files- gradio_demo/app.py +5 -4
gradio_demo/app.py
CHANGED
@@ -166,10 +166,11 @@ def instantir_restore(
|
|
166 |
else:
|
167 |
lq = [resize_img(lq.convert("RGB"), size=None)]
|
168 |
|
169 |
-
if cpu_offload:
|
170 |
-
|
171 |
-
else:
|
172 |
-
|
|
|
173 |
timesteps = [
|
174 |
i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
|
175 |
]
|
|
|
166 |
else:
|
167 |
lq = [resize_img(lq.convert("RGB"), size=None)]
|
168 |
|
169 |
+
#if cpu_offload:
|
170 |
+
# generator = torch.Generator().manual_seed(seed)
|
171 |
+
#else:
|
172 |
+
# generator = torch.Generator(device=device).manual_seed(seed)
|
173 |
+
generator = torch.Generator(device=device).manual_seed(seed)
|
174 |
timesteps = [
|
175 |
i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
|
176 |
]
|