remove dynamo disable

Author: a-r-r-o-w

src/diffusers/models/hooks.py

@@ -139,15 +139,13 @@ def init_hook(self, module: torch.nn.Module):
         module.to(dtype=self.storage_dtype)
         return module
 
-    @torch._dynamo.disable(recursive=False)
     def pre_forward(self, module: torch.nn.Module, *args, **kwargs):
         module.to(dtype=self.compute_dtype)
         # How do we account for LongTensor, BoolTensor, etc.?
         # args = tuple(align_maybe_tensor_dtype(arg, self.compute_dtype) for arg in args)
         # kwargs = {k: align_maybe_tensor_dtype(v, self.compute_dtype) for k, v in kwargs.items()}
         return args, kwargs
 
-    @torch._dynamo.disable(recursive=False)
     def post_forward(self, module: torch.nn.Module, output):
         module.to(dtype=self.storage_dtype)
         return output