lewtun HF staff commited on
Commit
83a0389
·
verified ·
1 Parent(s): 9354102

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import torch
8
  import spaces
9
  import os
10
 
11
- MODEL_ID = os.get("MODEL_ID", "llava-hf/llava-v1.6-mistral-7b-hf")
12
- REVISION = os.get("MODEL_REVISION", "main")
13
  processor = LlavaNextProcessor.from_pretrained(MODEL_ID, revision=REVISION)
14
 
15
  model = LlavaNextForConditionalGeneration.from_pretrained(MODEL_ID, revision=REVISION, torch_dtype=torch.float16, low_cpu_mem_usage=True)
 
8
  import spaces
9
  import os
10
 
11
+ MODEL_ID = os.environ.get("MODEL_ID", "llava-hf/llava-v1.6-mistral-7b-hf")
12
+ REVISION = os.environ.get("MODEL_REVISION", "main")
13
  processor = LlavaNextProcessor.from_pretrained(MODEL_ID, revision=REVISION)
14
 
15
  model = LlavaNextForConditionalGeneration.from_pretrained(MODEL_ID, revision=REVISION, torch_dtype=torch.float16, low_cpu_mem_usage=True)