profaker commited on
Commit
e711365
·
verified ·
1 Parent(s): 88017f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
71
  guidance_scale=guidance_scale,
72
  num_inference_steps=int(steps),
73
  strength=strength,
74
- generator=torch.Generator(device=device).manual_seed(42),
75
  clip_skip=1
76
  )
77
  inpainted_image = output.images[0]
 
71
  guidance_scale=guidance_scale,
72
  num_inference_steps=int(steps),
73
  strength=strength,
74
+ generator=torch.Generator(device="cuda").manual_seed(42),
75
  clip_skip=1
76
  )
77
  inpainted_image = output.images[0]