Skip to content

Commit d8b5f72

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

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

invokeai/backend/model_manager/legacy_probe.py

+2
Original file line numberDiff line numberDiff line change
@@ -569,6 +569,8 @@ def get_variant_type(self) -> ModelVariantType:
569569

570570
if in_channels is None:
571571
# If we cannot find the in_channels, we assume that this is a normal variant. Log a warning.
572+
# If this occurs, we should add a test case for the affected model here:
573+
# tests/backend/flux/test_flux_state_dict_utils.py
572574
logger.warning(
573575
f"{self.model_path} does not have img_in.weight or model.diffusion_model.img_in.weight key. Assuming normal variant."
574576
)

0 commit comments

Comments
 (0)