Nyandwi commited on
Commit
8f0f630
·
verified ·
1 Parent(s): 68c0ca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -589,7 +589,7 @@ if __name__ == "__main__":
589
 
590
  model_path = args.model_path
591
  filt_invalid = "cut"
592
- model_name = "pangea-llava-qwen"
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
 
589
 
590
  model_path = args.model_path
591
  filt_invalid = "cut"
592
+ model_name = "Pangea-7B-qwen"
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