Vision-CAIR commited on
Commit
0733f19
1 Parent(s): bc559f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ class Chat:
68
  self.version = "qwen"
69
  model_name = "cambrian_qwen"
70
  model_path = snapshot_download("Vision-CAIR/LongVU_Qwen2_7B", repo_type="model")
71
- + device = "cuda"
72
 
73
  self.tokenizer, self.model, self.processor, _ = load_pretrained_model(model_path, None, model_name, device=device)
74
  self.model.eval()
 
68
  self.version = "qwen"
69
  model_name = "cambrian_qwen"
70
  model_path = snapshot_download("Vision-CAIR/LongVU_Qwen2_7B", repo_type="model")
71
+ device = "cuda"
72
 
73
  self.tokenizer, self.model, self.processor, _ = load_pretrained_model(model_path, None, model_name, device=device)
74
  self.model.eval()