Spaces:

darpanaswal
/

Patent_Retrieval

Configuration error

darpanaswal commited on Apr 11

Commit

19561c9

verified ·

1 Parent(s): 58f39f1

Update cross_encoder_reranking_train.py

Files changed (1) hide show

cross_encoder_reranking_train.py CHANGED Viewed

@@ -13,7 +13,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
-embedder = SentenceTransformer("all-MiniLM-L12-v2").to(device)
 # embedder = SentenceTransformer("AI-Growth-Lab/PatentSBERTa").to(device)

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
+embedder = SentenceTransformer("all-MiniLM-L6-v2").to(device)
 # embedder = SentenceTransformer("AI-Growth-Lab/PatentSBERTa").to(device)