diff --git a/iopaint/api.py b/iopaint/api.py index 29b4f929..363d23df 100644 --- a/iopaint/api.py +++ b/iopaint/api.py @@ -16,6 +16,7 @@ torch._C._jit_override_can_fuse_on_gpu(False) torch._C._jit_set_texpr_fuser_enabled(False) torch._C._jit_set_nvfuser_enabled(False) + torch._C._jit_set_profiling_mode(False) except: pass