From b88cb9a344d5f8601dbb9e9af13c9d9c4b466112 Mon Sep 17 00:00:00 2001 From: blepping Date: Wed, 28 Aug 2024 12:50:39 -0600 Subject: [PATCH] Remove require_prefix in gguf_sd_loader --- nodes.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index d702e0c..609716e 100644 --- a/nodes.py +++ b/nodes.py @@ -32,7 +32,7 @@ def gguf_sd_loader_get_orig_shape(reader, tensor_name): raise TypeError(f"Bad original shape metadata for {field_key}: Expected ARRAY of INT32, got {field.types}") return torch.Size(tuple(int(field.parts[part_idx][0]) for part_idx in field.data)) -def gguf_sd_loader(path, handle_prefix="model.diffusion_model.", require_prefix=False): +def gguf_sd_loader(path, handle_prefix="model.diffusion_model."): """ Read state dict as fake tensors """ @@ -54,8 +54,6 @@ def gguf_sd_loader(path, handle_prefix="model.diffusion_model.", require_prefix= has_prefix = any(s.startswith(handle_prefix) for s in tensor_names) else: has_prefix = False - if require_prefix and not has_prefix: - raise RuntimeError(f"Loader requires prefix {handle_prefix!r} which does not exist in the model") for tensor in reader.tensors: sd_key = tensor_name = tensor.name if has_prefix: