We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1a1734d commit d8b5f72Copy full SHA for d8b5f72
invokeai/backend/model_manager/legacy_probe.py
@@ -569,6 +569,8 @@ def get_variant_type(self) -> ModelVariantType:
569
570
if in_channels is None:
571
# 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
574
logger.warning(
575
f"{self.model_path} does not have img_in.weight or model.diffusion_model.img_in.weight key. Assuming normal variant."
576
)
0 commit comments