diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py index 0423b7287193..84baf1af1023 100644 --- a/src/diffusers/models/modeling_utils.py +++ b/src/diffusers/models/modeling_utils.py @@ -199,10 +199,10 @@ def find_tensor_attributes(module: nn.Module) -> list[tuple[str, Tensor]]: gen = parameter._named_members(get_members_fn=find_tensor_attributes) last_tuple = None - for tuple in gen: - last_tuple = tuple - if tuple[1].is_floating_point(): - return tuple[1].dtype + for named_tensor in gen: + last_tuple = named_tensor + if named_tensor[1].is_floating_point(): + return named_tensor[1].dtype if last_tuple is not None: # fallback to the last dtype