From e6af602c0d59acbb8b7dbbf018f0d7eb2adf64c0 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 27 Oct 2025 21:40:18 +0300 Subject: [PATCH] handle files = str case --- modules/sdnq/file_loader.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/sdnq/file_loader.py b/modules/sdnq/file_loader.py index 5a5796e6b..42edba0f9 100644 --- a/modules/sdnq/file_loader.py +++ b/modules/sdnq/file_loader.py @@ -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: