1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Add Serialized Type Name kwarg in Model Output (#10502)

* Update outputs.py
This commit is contained in:
Aamir Nazir
2025-04-21 22:45:28 +04:00
committed by GitHub
parent 79ea8eb258
commit aff574fb29

View File

@@ -71,6 +71,7 @@ class BaseOutput(OrderedDict):
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: