torch._C._jit_set_profiling_mode(False)

Sanster · Nov 23, 2024 · e9264fa · e9264fa
1 parent 074fa55
commit e9264fa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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