Tevatron
/

OmniEmbed-v0.1

Visual Document Retrieval

Model card Files Files and versions

ArvinZhuang commited on May 26

Commit

1159ba9

·

verified ·

1 Parent(s): 37eb4a2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B")
 model = Qwen2_5OmniThinkerForConditionalGeneration.from_pretrained(
-    'ArvinZhuang/OmniEmbed-test',
     attn_implementation="flash_attention_2",
     torch_dtype=torch.bfloat16
 ).to(device).eval()

 processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B")
 model = Qwen2_5OmniThinkerForConditionalGeneration.from_pretrained(
+    'Tevatron/OmniEmbed-v0.1',
     attn_implementation="flash_attention_2",
     torch_dtype=torch.bfloat16
 ).to(device).eval()