mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-14 15:07:57 -05:00
Delete all flux bundle state dict keys when extracting the transformer state dict
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user