Don't reset dynamo on upcaler compile

Artheriax · Dec 31, 2024 · 8f35f09 · 8f35f09
1 parent 945c57c
commit 8f35f09
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/modules/upscaler.py b/modules/upscaler.py
@@ -217,7 +217,6 @@ def compile_upscaler(model):
     try:
         if "Upscaler" in shared.opts.cuda_compile and shared.opts.cuda_compile_backend != 'none':
             import torch._dynamo # pylint: disable=unused-import,redefined-outer-name
-            torch._dynamo.reset() # pylint: disable=protected-access
             if shared.opts.cuda_compile_backend not in torch._dynamo.list_backends(): # pylint: disable=protected-access
                 shared.log.warning(f"Upscaler compile not available: backend={shared.opts.cuda_compile_backend} available={torch._dynamo.list_backends()}") # pylint: disable=protected-access
                 return model