openbmb
/

VisRAG-Ret

Feature Extraction

Model card Files Files and versions Community

tcy6 commited on 3 days ago

Commit

e7c5b30

•

1 Parent(s): 890393c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -89,7 +89,7 @@ def encode(text_or_image_list):
     return embeddings
 tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
-model = AutoModel.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
 model.eval()
 script_dir = os.path.dirname(os.path.realpath(__file__))

     return embeddings
 tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
+model = AutoModel.from_pretrained("openbmb/VisRAG", torch_dtype=torch.bfloat16, trust_remote_code=True)
 model.eval()
 script_dir = os.path.dirname(os.path.realpath(__file__))