From 7dc6329bb72974dc10bd005dd134b28cf43b15a3 Mon Sep 17 00:00:00 2001 From: Adam Hendel Date: Wed, 24 Jan 2024 21:26:44 -0500 Subject: [PATCH] fmt (#15) --- app/models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/app/models.py b/app/models.py index 817707d..c31a035 100644 --- 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: