diff --git a/loader.py b/loader.py index 8e7dd81..7dcbd6c 100644 --- a/loader.py +++ b/loader.py @@ -7,7 +7,7 @@ import gguf from .ops import GGMLTensor from .dequant import is_quantized, dequantize_tensor -IMG_ARCH_LIST = {"flux", "sd1", "sdxl", "sd3", "aura", "hidream", "ltxv", "hyvid", "wan"} +IMG_ARCH_LIST = {"flux", "sd1", "sdxl", "sd3", "aura", "hidream", "ltxv", "hyvid", "wan", "pig"} TXT_ARCH_LIST = {"t5", "t5encoder", "llama"} def get_orig_shape(reader, tensor_name): @@ -70,7 +70,7 @@ def gguf_sd_loader(path, handle_prefix="model.diffusion_model.", return_arch=Fal # detect and verify architecture compat = None arch_str = get_field(reader, "general.architecture", str) - if arch_str in [None, "pig"]: + if arch_str in [None]: if is_text_model: raise ValueError(f"This text model is incompatible with llama.cpp!\nConsider using the safetensors version\n({path})") compat = "sd.cpp" if arch_str is None else arch_str