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