From 16df548b777cf8f7e3de1389f02d96f607938161 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Tue, 12 Aug 2025 09:20:08 +0800
Subject: [PATCH] fix: expand supported file extensions in CheckpointScanner
 initialization, fixes #353

---
 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 c81be7ef..3f298156 100644
--- a/py/services/checkpoint_scanner.py
+++ b/py/services/checkpoint_scanner.py
@@ -13,7 +13,7 @@ class CheckpointScanner(ModelScanner):
     
     def __init__(self):
         # Define supported file extensions
-        file_extensions = {'.safetensors', '.ckpt', '.pt', '.pth', '.sft', '.gguf'}
+        file_extensions = {'.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft', '.gguf'}
         super().__init__(
             model_type="checkpoint",
             model_class=CheckpointMetadata,