Merge branch 'dev' of https://github.com/vladmandic/sdnext into dev

2026-09-20 01:31:13 +02:00 · 2025-12-16 19:30:50 +01:00
parent f7a3fd15ca 0d45e851fb
commit 40a6cbfe00
1 changed files with 2 additions and 0 deletions
@@ -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)