Support non-quantized bundles

2026-04-23 03:00:31 -04:00 · 2024-09-03 16:04:42 -04:00
parent a005d06255
commit 2d0e34e57b
1 changed files with 2 additions and 0 deletions
--- a/invokeai/backend/model_manager/load/model_loaders/flux.py
+++ b/invokeai/backend/model_manager/load/model_loaders/flux.py
@@ -191,6 +191,8 @@ class FluxCheckpointModel(ModelLoader):
        with SilenceWarnings():
            model = Flux(params[config.config_path])
            sd = load_file(model_path)
+            if "model.diffusion_model.double_blocks.0.img_attn.norm.key_norm.scale" in sd:
+                sd = convert_bundle_to_flux_transformer_checkpoint(sd)
            model.load_state_dict(sd, assign=True)
        return model