fmt (#15)

tembo-io · Jan 25, 2024 · 7dc6329 · 7dc6329
1 parent cc5e770
commit 7dc6329
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/app/models.py b/app/models.py
@@ -41,6 +41,9 @@ def _model_dir(model: str) -> str:
 
 def model_org_name(model_name: str) -> str:
     """prepends with the HF if the org is not specified"""
+    if model_name == "all_MiniLM_L12_v2":
+        model_name = "all-MiniLM-L12-v2"
+
     if "/" not in model_name:
         return f"{_HF_ORG}/{model_name}"
     else: