Nyandwi commited on
Commit
3639a12
·
verified ·
1 Parent(s): b88b6ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- #model=model.to(torch.device('cuda'))
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()