Hugging Face's logo

openbmb
/
VisRAG-Ret

Feature Extraction
Safetensors
English
minicpmv
VisRAG
custom_code
Model card Files Community
tcy6 commited on
Commit
e7c5b30
·
verified ·
1 Parent(s): 890393c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -89,7 +89,7 @@ def encode(text_or_image_list):
89
  return embeddings
90
 
91
  tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
92
- model = AutoModel.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
93
  model.eval()
94
 
95
  script_dir = os.path.dirname(os.path.realpath(__file__))
 
89
  return embeddings
90
 
91
  tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
92
+ model = AutoModel.from_pretrained("openbmb/VisRAG", torch_dtype=torch.bfloat16, trust_remote_code=True)
93
  model.eval()
94
 
95
  script_dir = os.path.dirname(os.path.realpath(__file__))