Hugging Face's logo

Spaces:
cbensimon
/
SALMONN-7B-gradio-zero
Paused

App Files Community
cbensimon HF Staff commited on
Commit
ff3e3b6
·
verified ·
1 Parent(s): b85bc94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -42,6 +42,7 @@ model = SALMONN(
42
  vicuna_path=args.vicuna_path,
43
  low_resource=args.low_resource,
44
  lora_alpha=28,
 
45
  )
46
  model.to(args.device)
47
  model.eval()
 
42
  vicuna_path=args.vicuna_path,
43
  low_resource=args.low_resource,
44
  lora_alpha=28,
45
+ device='cpu',
46
  )
47
  model.to(args.device)
48
  model.eval()