Support non-quantized bundles

invoke-ai · Sep 3, 2024 · f4abfd2 · f4abfd2
1 parent 10f0ff7
commit f4abfd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/invokeai/backend/model_manager/load/model_loaders/flux.py b/invokeai/backend/model_manager/load/model_loaders/flux.py
@@ -191,6 +191,8 @@ def _load_from_singlefile(
         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