From d080379e9488ebe7e41bee4e5cc7fa3da4826125 Mon Sep 17 00:00:00 2001
From: Aryan <aryan@huggingface.co>
Date: Fri, 21 Feb 2025 08:41:11 +0100
Subject: [PATCH] try fix for tests

---
 src/diffusers/hooks/layerwise_casting.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/diffusers/hooks/layerwise_casting.py b/src/diffusers/hooks/layerwise_casting.py
index 6f2cfdc348..ff0be07cf5 100644
--- a/src/diffusers/hooks/layerwise_casting.py
+++ b/src/diffusers/hooks/layerwise_casting.py
@@ -52,6 +52,8 @@ class LayerwiseCastingHook(ModelHook):
     _is_stateful = False
 
     def __init__(self, storage_dtype: torch.dtype, compute_dtype: torch.dtype, non_blocking: bool) -> None:
+        super().__init__()
+
         self.storage_dtype = storage_dtype
         self.compute_dtype = compute_dtype
         self.non_blocking = non_blocking