vec-ai
/

lychee-rerank-mm

Model card Files Files and versions

DDDDZQ commited on Oct 17

Commit

130c3f1

·

verified ·

1 Parent(s): 6d5f4f5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -122,7 +122,7 @@ def compute_logits(inputs, **kwargs):
     return scores
-model_name_or_path = "vec-ai/lychee-rerank-mm-3b"
 min_pixels = 4*28*28
 max_pixels = 1280*28*28
 tokenizer = AutoProcessor.from_pretrained(model_name_or_path, padding_side='left', min_pixels=min_pixels, max_pixels=max_pixels, trust_remote_code=True)

     return scores
+model_name_or_path = "vec-ai/lychee-rerank-mm"
 min_pixels = 4*28*28
 max_pixels = 1280*28*28
 tokenizer = AutoProcessor.from_pretrained(model_name_or_path, padding_side='left', min_pixels=min_pixels, max_pixels=max_pixels, trust_remote_code=True)