Add support for '.gguf' file extension in CheckpointScanner

This commit is contained in:
Will Miao
2025-04-10 21:15:12 +08:00
parent 131c3cc324
commit 311bf1f157

View File

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