From ca715a9771fd87dbc5f7d7a4397c755fd6ed0b07 Mon Sep 17 00:00:00 2001
From: Aryan <aryan@huggingface.co>
Date: Mon, 14 Apr 2025 18:04:13 +0530
Subject: [PATCH] Update src/diffusers/hooks/hooks.py

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
---
 src/diffusers/hooks/hooks.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/diffusers/hooks/hooks.py b/src/diffusers/hooks/hooks.py
index c42592783d..17be685874 100644
--- a/src/diffusers/hooks/hooks.py
+++ b/src/diffusers/hooks/hooks.py
@@ -58,14 +58,15 @@ class BaseMarkedState(BaseState):
         self._mark_name = None
 
     def __getattribute__(self, name):
-        if name in (
+        direct_attrs = (
             "get_current_state",
             "mark_state",
             "reset",
             "_init_args",
             "_init_kwargs",
             "_mark_name",
-            "_state_cache",
+            "_state_cache",)
+        if name in direct_attrs or _is_dunder_method(name):
         ) or _is_dunder_method(name):
             return object.__getattribute__(self, name)
         else: