bedead commited on
Commit
1496374
·
verified ·
1 Parent(s): 7d448bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ clipaway = CLIPAway(
21
  alpha_clip_path=config.alpha_clip_ckpt_pth,
22
  config=config,
23
  alpha_clip_id=config.alpha_clip_id,
24
- device="cuda",
25
  num_tokens=4
26
  )
27
 
 
21
  alpha_clip_path=config.alpha_clip_ckpt_pth,
22
  config=config,
23
  alpha_clip_id=config.alpha_clip_id,
24
+ device="cpu",
25
  num_tokens=4
26
  )
27