🔧 Remove prefix from name

shroominic · Feb 15, 2024 · cb5b77f · cb5b77f
1 parent ab8a9ea
commit cb5b77f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/funcchain/model/defaults.py b/src/funcchain/model/defaults.py
@@ -25,6 +25,7 @@ def get_gguf_model(
     """
     from huggingface_hub import hf_hub_download
 
+    name = name.removeprefix("TheBloke/")
     name = name.removesuffix("-GGUF")
     label = "Q5_K_M" if label == "latest" else label