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