Update invokeai/backend/model_manager/probe.py

Co-authored-by: Ryan Dick <[email protected]>
invoke-ai · Sep 3, 2024 · 9e55b7f · 9e55b7f
1 parent f4abfd2
commit 9e55b7f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py
@@ -229,7 +229,10 @@ def get_model_type_from_checkpoint(cls, model_path: Path, checkpoint: Optional[C
                     "cond_stage_model.",
                     "first_stage_model.",
                     "model.diffusion_model.",
+                    # FLUX models in the official BFL format contain keys with the "double_blocks." prefix.
                     "double_blocks.",
+                    # Some FLUX checkpoint files contain transformer keys prefixed with "model.diffusion_model".
+                    # This prefix is typically used to distinguish between multiple models bundled in a single file.
                     "model.diffusion_model.double_blocks.",
                 )
             ):