Skip to content

Commit e37fba0

Browse files
committed
Delete the model prefix specified by the user
1 parent ed46c01 commit e37fba0

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

evaluation/scripts/locomo/locomo_eval.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434

3535

3636
try:
37-
sentence_model_name = "/home/liupeng/dataroot/models/Qwen/Qwen3-Embedding-0.6B"
37+
sentence_model_name = "Qwen/Qwen3-Embedding-0.6B"
3838
sentence_model = SentenceTransformer(sentence_model_name)
3939
print(f"SentenceTransformer model : {sentence_model_name} loaded successfully.")
4040
except Exception as e:
@@ -138,9 +138,7 @@ def calculate_semantic_similarity(gold_answer, response):
138138

139139
try:
140140
if sentence_model is None:
141-
sentence_model = SentenceTransformer(
142-
"/home/liupeng/dataroot/models/Qwen/Qwen3-Embedding-0.6B"
143-
)
141+
sentence_model = SentenceTransformer("Qwen/Qwen3-Embedding-0.6B")
144142

145143
gold_embedding = sentence_model.encode([gold_answer], show_progress_bar=False)[0]
146144
response_embedding = sentence_model.encode([response], show_progress_bar=False)[0]

0 commit comments

Comments
 (0)