diff --git a/modules/sdnq/file_loader.py b/modules/sdnq/file_loader.py index 7ea8e1fbe..5e921da0d 100644 --- a/modules/sdnq/file_loader.py +++ b/modules/sdnq/file_loader.py @@ -19,7 +19,7 @@ def load_safetensors(files: list[str], state_dict: dict | None = None, key_mappi state_dict = {} for fn in files: with safe_open(fn, framework="pt", device=str(device)) as f: - for key in f: + for key in f.keys(): # safe_open exposes keys() but is not iterable state_dict[map_keys(key, key_mapping)] = f.get_tensor(key)