credent007 commited on
Commit
81767fa
·
verified ·
1 Parent(s): 0eb7146

Update model_loader.py

Browse files
Files changed (1) hide show
  1. model_loader.py +4 -1
model_loader.py CHANGED
@@ -6,7 +6,10 @@ MODEL_NAME = "Qwen/Qwen2.5-VL-7B-Instruct"
6
  device = "cuda" if torch.cuda.is_available() else "cpu"
7
 
8
  print("Loading processor...")
9
- processor = AutoProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)
 
 
 
10
 
11
  print("Loading model...")
12
  model = AutoModelForVision2Seq.from_pretrained(
 
6
  device = "cuda" if torch.cuda.is_available() else "cpu"
7
 
8
  print("Loading processor...")
9
+ processor = AutoProcessor.from_pretrained(
10
+ MODEL_NAME,
11
+ trust_remote_code=True,
12
+ use_fast=True) # use_fast to avoid warnings in logs
13
 
14
  print("Loading model...")
15
  model = AutoModelForVision2Seq.from_pretrained(