Nyandwi commited on
Commit
d4bca32
·
verified ·
1 Parent(s): a95b90a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -726,8 +726,9 @@ if __name__ == "__main__":
726
  model_path = args.model_path
727
  filt_invalid = "cut"
728
  args = {"multimodal": True}
729
- model_name = get_model_name_from_path(args.model_path)
730
- tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit, **args)
 
731
  model=model.to(torch.device('cuda'))
732
  chat_image_num = 0
733
  demo.launch()
 
726
  model_path = args.model_path
727
  filt_invalid = "cut"
728
  args = {"multimodal": True}
729
+ #model_name = get_model_name_from_path(args.model_path)
730
+ model_name = "pangea_llava_qwen"
731
+ tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, None, model_name, args.load_8bit, args.load_4bit, **args)
732
  model=model.to(torch.device('cuda'))
733
  chat_image_num = 0
734
  demo.launch()