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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user