From 0d45e851fbbdf42ad7ca035de82a34b0c5120374 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 16 Dec 2025 16:43:43 +0300 Subject: [PATCH] disable compile on balanced offload hooks --- modules/sd_offload.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/sd_offload.py b/modules/sd_offload.py index 6c727fe04..721716bde 100644 --- a/modules/sd_offload.py +++ b/modules/sd_offload.py @@ -213,6 +213,7 @@ class OffloadHook(accelerate.hooks.ModelHook): return False return True + @torch.compiler.disable def pre_forward(self, module, *args, **kwargs): _id = id(module) @@ -261,6 +262,7 @@ class OffloadHook(accelerate.hooks.ModelHook): self.last_pre = _id return args, kwargs + @torch.compiler.disable def post_forward(self, module, output): if self.last_post != id(module): self.last_post = id(module)