Spaces:
Starting
Starting
Update app.py
Browse files
app.py
CHANGED
@@ -591,6 +591,6 @@ if __name__ == "__main__":
|
|
591 |
filt_invalid = "cut"
|
592 |
model_name = get_model_name_from_path(args.model_path)
|
593 |
tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
|
594 |
-
model=model.to(torch.device('cuda'))
|
595 |
chat_image_num = 0
|
596 |
demo.launch()
|
|
|
591 |
filt_invalid = "cut"
|
592 |
model_name = get_model_name_from_path(args.model_path)
|
593 |
tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit)
|
594 |
+
#model=model.to(torch.device('cuda'))
|
595 |
chat_image_num = 0
|
596 |
demo.launch()
|