JOY-Huang commited on
Commit
c4c4126
·
1 Parent(s): 10fca91
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -157,7 +157,8 @@ def instantir_restore(
157
  elif preview_start > 1.0:
158
  preview_start = preview_start / steps
159
 
160
- lq, out_size = [resize_img(lq, width=width, height=height)]
 
161
  generator = torch.Generator(device=device).manual_seed(seed)
162
  timesteps = [
163
  i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
 
157
  elif preview_start > 1.0:
158
  preview_start = preview_start / steps
159
 
160
+ lq, out_size = resize_img(lq, width=width, height=height)
161
+ lq = [lq]
162
  generator = torch.Generator(device=device).manual_seed(seed)
163
  timesteps = [
164
  i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)