vidore
/

colqwen2-v0.1

Visual Document Retrieval

vidore-experimental

Model card Files Files and versions

tonywu71 commited on Oct 7, 2024

Commit

11e4fe7

·

verified ·

1 Parent(s): b87788f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ model = ColQwen2.from_pretrained(
         "vidore/colqwen2-v0.1",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",  # or "mps" if on Apple Silicon
-    )
 processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
 # Your inputs

         "vidore/colqwen2-v0.1",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",  # or "mps" if on Apple Silicon
+    ).eval()
 processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
 # Your inputs