Skip to content

Commit 042f0e4

Browse files
authored
Update README.md
1 parent 46062d9 commit 042f0e4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

FlagEmbedding/visual/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ import torch
6464
from FlagEmbedding.visual.modeling import Visualized_BGE
6565

6666
model = Visualized_BGE(model_name_bge = "BAAI/bge-base-en-v1.5", model_weight="path: Visualized_base_en_v1.5.pth")
67-
67+
model.eval()
6868
with torch.no_grad():
6969
query_emb = model.encode(text="Are there sidewalks on both sides of the Mid-Hudson Bridge?")
7070
candi_emb_1 = model.encode(text="The Mid-Hudson Bridge, spanning the Hudson River between Poughkeepsie and Highland.", image="./imgs/wiki_candi_1.jpg")

0 commit comments

Comments
 (0)