From da72554f586aa43da943c25ed6438ba46535bb78 Mon Sep 17 00:00:00 2001
From: teleprint-me <77757836+teleprint-me@users.noreply.github.com>
Date: Sat, 25 May 2024 19:11:56 -0400
Subject: [PATCH] feat: Add static methods for resolving model types and model
 extensions

---
 gguf-py/gguf/huggingface_hub.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/gguf-py/gguf/huggingface_hub.py b/gguf-py/gguf/huggingface_hub.py
index 1026ce17d..53469835d 100644
--- a/gguf-py/gguf/huggingface_hub.py
+++ b/gguf-py/gguf/huggingface_hub.py
@@ -219,6 +219,22 @@ class HFHubModel(HFHubBase):
         self._request = HFHubRequest(auth_token, model_path, logger)
         self._tokenizer = HFHubTokenizer(model_path, logger)
 
+    @staticmethod
+    def get_model_type_name(model_type: ModelFileType) -> str:
+        return MODEL_FILE_TYPE_NAMES.get(model_type, "")
+
+    @staticmethod
+    def get_model_type(model_name: str) -> ModelFileType:
+        return {
+            ".pt": ModelFileType.PT,
+            ".pth": ModelFileType.PTH,
+            ".bin": ModelFileType.BIN,
+            ".safetensors": ModelFileType.SAFETENSORS,
+            ".json": ModelFileType.JSON,
+            ".model": ModelFileType.MODEL,
+            ".gguf": ModelFileType.GGUF,
+        }.get(model_name, ModelFileType.NON)
+
     @property
     def request(self) -> HFHubRequest:
         return self._request