anhartit-chetan commited on
Commit
de423bd
·
1 Parent(s): e594f31

Change llava model again

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import torch
5
 
6
  # Load model & processor
7
  # model_id = "llava-hf/llava-1.5-7b-hf" # Exceeding 16 GB Memory
8
- model_id = "llava-hf/llava-1.5-7b-hf-int4"
9
-
10
  processor = AutoProcessor.from_pretrained(model_id)
11
  model = LlavaForConditionalGeneration.from_pretrained(
12
  model_id,
 
5
 
6
  # Load model & processor
7
  # model_id = "llava-hf/llava-1.5-7b-hf" # Exceeding 16 GB Memory
8
+ # model_id = "llava-hf/llava-1.5-7b-hf-int4" #Does not exists
9
+ model_id = "bczhou/tiny-llava-v1-hf"
10
  processor = AutoProcessor.from_pretrained(model_id)
11
  model = LlavaForConditionalGeneration.from_pretrained(
12
  model_id,