Update outputs.py

anzr299 · web-flow · commit f8b292d67da5 · 2025-01-08T22:13:21.000+04:00
diff --git a/src/diffusers/utils/outputs.py b/src/diffusers/utils/outputs.py
@@ -65,12 +65,14 @@ def __init_subclass__(cls) -> None:
                     cls,
                     torch.utils._pytree._dict_flatten,
                     lambda values, context: cls(**torch.utils._pytree._dict_unflatten(values, context)),
+                    serialized_type_name=f"{cls.__module__}.{cls.__name__}"
                 )
             else:
                 torch.utils._pytree.register_pytree_node(
                     cls,
                     torch.utils._pytree._dict_flatten,
                     lambda values, context: cls(**torch.utils._pytree._dict_unflatten(values, context)),
+                    serialized_type_name=f"{cls.__module__}.{cls.__name__}"
                 )
 
     def __post_init__(self) -> None:

Original file line number	Diff line number	Diff line change
`@@ -65,12 +65,14 @@ def __init_subclass__(cls) -> None:`
`65`	`65`	`cls,`
`66`	`66`	`torch.utils._pytree._dict_flatten,`
`67`	`67`	`lambda values, context: cls(**torch.utils._pytree._dict_unflatten(values, context)),`
	`68`	`+ serialized_type_name=f"{cls.__module__}.{cls.__name__}"`
`68`	`69`	`)`
`69`	`70`	`else:`
`70`	`71`	`torch.utils._pytree.register_pytree_node(`
`71`	`72`	`cls,`
`72`	`73`	`torch.utils._pytree._dict_flatten,`
`73`	`74`	`lambda values, context: cls(**torch.utils._pytree._dict_unflatten(values, context)),`
	`75`	`+ serialized_type_name=f"{cls.__module__}.{cls.__name__}"`
`74`	`76`	`)`
`75`	`77`
`76`	`78`	`def __post_init__(self) -> None:`