TurkishCodeMan commited on
Commit
70ba15f
·
verified ·
1 Parent(s): 367ec9f

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. models/model_loader.py +2 -2
models/model_loader.py CHANGED
@@ -44,9 +44,9 @@ def load_rerank_model(model_path: str = "nvidia/llama-nemotron-rerank-vl-1b-v2")
44
 
45
  processor = AutoProcessor.from_pretrained(
46
  model_path,
47
- trust_remote_code=True,
48
  max_input_tiles=6,
49
- use_thumbnail=True,""
50
  rerank_max_length=2048
51
  )
52
 
 
44
 
45
  processor = AutoProcessor.from_pretrained(
46
  model_path,
47
+ trust_remote_code=False,
48
  max_input_tiles=6,
49
+ use_thumbnail=True,
50
  rerank_max_length=2048
51
  )
52