Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -721,6 +721,8 @@ if __name__ == "__main__":
|
|
721 |
argparser.add_argument("--load-8bit", action="store_true")
|
722 |
argparser.add_argument("--load-4bit", action="store_true")
|
723 |
argparser.add_argument("--debug", action="store_true")
|
|
|
|
|
724 |
|
725 |
args = argparser.parse_args()
|
726 |
|
@@ -730,6 +732,6 @@ if __name__ == "__main__":
|
|
730 |
#model_name = get_model_name_from_path(args.model_path)
|
731 |
model_name = "pangea_llava_qwen"
|
732 |
tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, args.load_8bit, args.load_4bit, **args)
|
733 |
-
|
734 |
chat_image_num = 0
|
735 |
demo.launch()
|
|
|
721 |
argparser.add_argument("--load-8bit", action="store_true")
|
722 |
argparser.add_argument("--load-4bit", action="store_true")
|
723 |
argparser.add_argument("--debug", action="store_true")
|
724 |
+
argparser.add_argument("--multimodal", action="store_true")
|
725 |
+
|
726 |
|
727 |
args = argparser.parse_args()
|
728 |
|
|
|
732 |
#model_name = get_model_name_from_path(args.model_path)
|
733 |
model_name = "pangea_llava_qwen"
|
734 |
tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, args.load_8bit, args.load_4bit, **args)
|
735 |
+
model=model.to(torch.device('cuda'))
|
736 |
chat_image_num = 0
|
737 |
demo.launch()
|