Skip to content

Commit 3f81a8f

Browse files
docs: add reminder to FLUX variant probing to add test cases if we have a probe failure
1 parent d8d3d3a commit 3f81a8f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

invokeai/backend/model_manager/legacy_probe.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -572,6 +572,8 @@ def get_variant_type(self) -> ModelVariantType:
572572

573573
if in_channels is None:
574574
# If we cannot find the in_channels, we assume that this is a normal variant. Log a warning.
575+
# If this occurs, we should add a test case for the affected model here:
576+
# tests/backend/flux/test_flux_state_dict_utils.py
575577
logger.warning(
576578
f"{self.model_path} does not have img_in.weight or model.diffusion_model.img_in.weight key. Assuming normal variant."
577579
)

0 commit comments

Comments
 (0)