HuanjinYao commited on
Commit
43d70a9
1 Parent(s): 359e303

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ from PIL import Image
19
 
20
  tokenizer = AutoTokenizer.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', use_fast=False)
21
  model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', low_cpu_mem_usage=True,torch_dtype=torch.float16)
22
- image_processor = model.get_vision_tower()
 
23
  if not vision_tower.is_loaded:
24
  vision_tower.load_model()
25
  vision_tower.to(device=device, dtype=torch.float16)
 
19
 
20
  tokenizer = AutoTokenizer.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', use_fast=False)
21
  model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', low_cpu_mem_usage=True,torch_dtype=torch.float16)
22
+
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)