Nyandwi commited on
Commit
5c1d6e8
·
verified ·
1 Parent(s): f21c48a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -708,7 +708,7 @@ if __name__ == "__main__":
708
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
709
  argparser.add_argument("--port", default="6123", type=str)
710
  argparser.add_argument(
711
- "--model_path", default="neulab/Pangea-7B", type=str
712
  )
713
  # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
714
  argparser.add_argument("--model-base", type=str, default=None)
@@ -728,7 +728,7 @@ if __name__ == "__main__":
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("neulab/Pangea-7B", 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()
 
708
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
709
  argparser.add_argument("--port", default="6123", type=str)
710
  argparser.add_argument(
711
+ "--model_path", default="neulab/CulturalPangea-7B", type=str
712
  )
713
  # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
714
  argparser.add_argument("--model-base", type=str, default=None)
 
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(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()