Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|