Delete all flux bundle state dict keys when extracting the transformer state dict

Author: brandonrising

invokeai/backend/model_manager/util/model_util.py

@@ -143,6 +143,7 @@ def convert_bundle_to_flux_transformer_checkpoint(
 
     for k, v in transformer_state_dict.items():
         if not k.startswith("model.diffusion_model"):
+            keys_to_remove.append(k)  # This can be removed in the future if we only want to delete transformer keys
             continue
         if k.endswith("scale"):
             # Scale math must be done at bfloat16 due to our current flux model