torch._C._jit_set_profiling_mode(False)

2024-11-23 19:17:48 +08:00
parent 074fa5513a
commit e9264fadb7
1 changed files with 1 additions and 0 deletions
--- a/iopaint/api.py
+++ b/iopaint/api.py
@@ -16,6 +16,7 @@ try:
    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