paralym commited on
Commit
2d8c1cd
1 Parent(s): 07b9bc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -531,6 +531,8 @@ if __name__ == "__main__":
531
  model_name = get_model_name_from_path(args.model_path)
532
  tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
533
  print("### image_processor",image_processor)
 
 
534
  model=model.to(torch.device('cuda'))
535
  our_chatbot = None
536
  demo.launch()
 
531
  model_name = get_model_name_from_path(args.model_path)
532
  tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
533
  print("### image_processor",image_processor)
534
+ print("### model",model)
535
+ print("### tokenzier",tokenizer)
536
  model=model.to(torch.device('cuda'))
537
  our_chatbot = None
538
  demo.launch()