grayphite commited on
Commit
5073095
·
verified ·
1 Parent(s): b969905

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,10 +8,10 @@ import json
8
  import time
9
 
10
  # Load processor and model
11
- processor = AutoProcessor.from_pretrained("llava-hf/llava-1.5-7b-hf")
12
 
13
  model = LlavaForConditionalGeneration.from_pretrained(
14
- "llava-hf/llava-1.5-7b-hf",
15
  torch_dtype=torch.float16,
16
  device_map="auto"
17
  )
 
8
  import time
9
 
10
  # Load processor and model
11
+ processor = AutoProcessor.from_pretrained("liuhaotian/llava-v1.5-7b")
12
 
13
  model = LlavaForConditionalGeneration.from_pretrained(
14
+ "liuhaotian/llava-v1.5-7b",
15
  torch_dtype=torch.float16,
16
  device_map="auto"
17
  )