wenhu commited on
Commit
92d3e6f
·
verified ·
1 Parent(s): bd67f62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -698,10 +698,7 @@ if __name__ == "__main__":
698
  argparser = argparse.ArgumentParser()
699
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
700
  argparser.add_argument("--port", default="6123", type=str)
701
- argparser.add_argument(
702
- "--model_path", default="TIGER-Lab/MAmmoTH-VL2", type=str
703
- )
704
- # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
705
  argparser.add_argument("--model-base", type=str, default=None)
706
  argparser.add_argument("--num-gpus", type=int, default=1)
707
  argparser.add_argument("--conv-mode", type=str, default=None)
@@ -720,4 +717,4 @@ if __name__ == "__main__":
720
  tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
721
  model=model.to(torch.device('cuda'))
722
  chat_image_num = 0
723
- demo.launch()
 
698
  argparser = argparse.ArgumentParser()
699
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
700
  argparser.add_argument("--port", default="6123", type=str)
701
+ argparser.add_argument("--model_path", default="TIGER-Lab/MAmmoTH-VL2", type=str)
 
 
 
702
  argparser.add_argument("--model-base", type=str, default=None)
703
  argparser.add_argument("--num-gpus", type=int, default=1)
704
  argparser.add_argument("--conv-mode", type=str, default=None)
 
717
  tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
718
  model=model.to(torch.device('cuda'))
719
  chat_image_num = 0
720
+ demo.launch(server_port=args.port)