Don't reset dynamo on upcaler compile

This commit is contained in:
Disty0
2024-12-31 19:56:03 +03:00
parent 945c57c433
commit 8f35f09117
-1
View File
@@ -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