Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
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(
|
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()
|