diff --git a/src/diffusers/utils/outputs.py b/src/diffusers/utils/outputs.py index 6080a86b871a..c625746cf8a9 100644 --- a/src/diffusers/utils/outputs.py +++ b/src/diffusers/utils/outputs.py @@ -71,6 +71,7 @@ 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__}", ) def __post_init__(self) -> None: