cocktailpeanut commited on
Commit
693604f
·
1 Parent(s): c12474d
Files changed (1) hide show
  1. 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
- generator = torch.Generator().manual_seed(seed)
171
- else:
172
- generator = torch.Generator(device=device).manual_seed(seed)
 
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
  ]