From aff574fb2968f83b59221eb0d8f9c0b7587f53d2 Mon Sep 17 00:00:00 2001
From: Aamir Nazir <aamir.nazir@intel.com>
Date: Mon, 21 Apr 2025 22:45:28 +0400
Subject: [PATCH] Add Serialized Type Name kwarg in Model Output (#10502)

* Update outputs.py
---
 src/diffusers/utils/outputs.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/utils/outputs.py b/src/diffusers/utils/outputs.py
index 6080a86b87..c625746cf8 100644
--- a/src/diffusers/utils/outputs.py
+++ b/src/diffusers/utils/outputs.py
@@ -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: