Unverified Commit aff574fb authored by Aamir Nazir's avatar Aamir Nazir Committed by GitHub
Browse files

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

* Update outputs.py
parent 79ea8eb2
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment