fix: expand supported file extensions in CheckpointScanner initialization, fixes #353

This commit is contained in:
Will Miao
2025-08-12 09:20:08 +08:00
parent 425c33ae00
commit 16df548b77

View File

@@ -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,