fffiloni commited on
Commit
40ea8f0
1 Parent(s): d9a10ae

Update app_with_diffusers.py

Browse files
Files changed (1) hide show
  1. app_with_diffusers.py +1 -1
app_with_diffusers.py CHANGED
@@ -80,7 +80,7 @@ def infer(prompt, input_image, steps=30, cfg_scale=7.0, guidance_end=1.0,
80
  low_quality_image = Image.open(input_image).convert("RGB")
81
 
82
  lq = [resize_img(low_quality_image, size=(width, height))]
83
- generator = torch.Generator(device=device).manual_seed(seed)
84
  timesteps = [
85
  i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
86
  ]
 
80
  low_quality_image = Image.open(input_image).convert("RGB")
81
 
82
  lq = [resize_img(low_quality_image, size=(width, height))]
83
+ generator = torch.Generator(device='cuda').manual_seed(seed)
84
  timesteps = [
85
  i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
86
  ]