Fix missing extension addition from upstream

This commit is contained in:
doctorpangloss 2024-04-29 13:32:46 -07:00
parent f965fb2bc0
commit 46a712b4d6

View File

@ -10,7 +10,7 @@ from pkg_resources import resource_filename
from ..cli_args import args
supported_pt_extensions = frozenset(['.ckpt', '.pt', '.bin', '.pth', '.safetensors'])
supported_pt_extensions = frozenset(['.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl'])
@dataclasses.dataclass