iitolstykh commited on
Commit
c480799
·
verified ·
1 Parent(s): 59b6d42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def load_pipeline():
40
  image_guidance_scale=1.2,
41
  guidance_scale=4.5,
42
  num_inference_steps=20,
43
- device="cuda:0",
44
  )
45
 
46
  return editor_pipeline
 
40
  image_guidance_scale=1.2,
41
  guidance_scale=4.5,
42
  num_inference_steps=20,
43
+ device="cuda",
44
  )
45
 
46
  return editor_pipeline