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)
|