From 311bf1f1575ec478311e072c1fd42d8dcf3d5ec8 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Thu, 10 Apr 2025 21:15:12 +0800
Subject: [PATCH] Add support for '.gguf' file extension in CheckpointScanner

---
 py/services/checkpoint_scanner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/services/checkpoint_scanner.py b/py/services/checkpoint_scanner.py
index 27d15273..4cc77b6a 100644
--- a/py/services/checkpoint_scanner.py
+++ b/py/services/checkpoint_scanner.py
@@ -25,7 +25,7 @@ class CheckpointScanner(ModelScanner):
     def __init__(self):
         if not hasattr(self, '_initialized'):
             # Define supported file extensions
-            file_extensions = {'.safetensors', '.ckpt', '.pt', '.pth', '.sft'}
+            file_extensions = {'.safetensors', '.ckpt', '.pt', '.pth', '.sft', '.gguf'}
             super().__init__(
                 model_type="checkpoint",
                 model_class=CheckpointMetadata,