Add custom artifact path to CoherenceScorer download

wandb · Feb 5, 2025 · a2a8cf4 · a2a8cf4
1 parent c97963d
commit a2a8cf4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/weave/scorers/coherence_scorer.py b/weave/scorers/coherence_scorer.py
@@ -35,6 +35,8 @@ def _load_pipeline(self) -> None:
         from transformers import pipeline
         if os.path.isdir(self.model_name_or_path):
             self._local_model_path = self.model_name_or_path
+        elif self.model_name_or_path != "":
+            self._local_model_path = download_model(self.model_name_or_path)
         else:
             self._local_model_path = download_model(MODEL_PATHS["coherence_scorer"])