handle files = str case

2026-09-19 17:24:32 +02:00 · 2025-10-27 21:40:18 +03:00
parent 9308a28c9d
commit e6af602c0d
1 changed files with 2 additions and 0 deletions
@@ -47,6 +47,8 @@ def load_streamer(files: list[str], state_dict: dict = None, key_mapping: dict =

 def load_files(files: list[str], state_dict: dict = None, key_mapping: dict = None, device: torch.device = "cpu", method: str = None) -> dict:
    # note: files is list-of-files within a module for chunked loading, not accross model
+    if isinstance(files, str):
+        files = [files]
    if method is None:
        method = 'safetensors'
    if state_dict is None: