Ahren09 commited on
Commit
27c7e48
·
verified ·
1 Parent(s): f8dc691

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ model = loaded_model_name = tokenizer = image_processor = context_len = my_gener
40
  def load_model_async(model_path, model_name):
41
  global tokenizer, model, image_processor, context_len, loaded_model_name, my_generator
42
  print(f"Loading {model_name} model ... ")
43
- tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name)
44
  if "llava" in model_name.lower():
45
  loaded_model_name = "LLaVA"
46
  else:
 
40
  def load_model_async(model_path, model_name):
41
  global tokenizer, model, image_processor, context_len, loaded_model_name, my_generator
42
  print(f"Loading {model_name} model ... ")
43
+ tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, load_4bit=True)
44
  if "llava" in model_name.lower():
45
  loaded_model_name = "LLaVA"
46
  else: