HuanjinYao commited on
Commit
8130fc5
1 Parent(s): 43d70a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B
23
  vision_tower = model.get_vision_tower()
24
  if not vision_tower.is_loaded:
25
  vision_tower.load_model()
26
- vision_tower.to(device=device, dtype=torch.float16)
27
  image_processor = vision_tower.image_processor
28
 
29
  model.to('cuda')
 
23
  vision_tower = model.get_vision_tower()
24
  if not vision_tower.is_loaded:
25
  vision_tower.load_model()
26
+ vision_tower.to(device='cuda', dtype=torch.float16)
27
  image_processor = vision_tower.image_processor
28
 
29
  model.to('cuda')