Fix loop var naming (#38885)

This commit is contained in:
Matt
2025-06-18 14:45:01 +01:00
committed by GitHub
parent 1fc67a25c6
commit 9cd7570f34

View File

@@ -380,10 +380,10 @@ def get_parameter_dtype(parameter: Union[nn.Module, "ModuleUtilsMixin"]):
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 gen_tuple in gen:
last_tuple = gen_tuple
if gen_tuple[1].is_floating_point():
return gen_tuple[1].dtype
if last_tuple is not None:
# fallback to the last dtype