stack86 commited on
Commit
7311c4c
·
verified ·
1 Parent(s): d41bb87

Update app.py

Browse files

change cuda to cpu

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def generate_image(prompt):
32
  prompt,
33
  num_inference_steps=1,
34
  guidance_scale=0,
35
- generator=torch.Generator(device="cuda").manual_seed(seed)
36
  )
37
  return results.images[0]
38
 
 
32
  prompt,
33
  num_inference_steps=1,
34
  guidance_scale=0,
35
+ generator=torch.Generator(device="cpu").manual_seed(seed)
36
  )
37
  return results.images[0]
38