csukuangfj commited on
Commit
0e6190d
1 Parent(s): 7767222

small fixes

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -309,6 +309,13 @@ with demo:
309
  )
310
  gr.Markdown(description)
311
 
 
 
 
 
 
 
 
312
  if __name__ == "__main__":
313
  formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
314
 
309
  )
310
  gr.Markdown(description)
311
 
312
+ torch.set_num_threads(1)
313
+ torch.set_num_interop_threads(1)
314
+
315
+ torch._C._jit_set_profiling_executor(False)
316
+ torch._C._jit_set_profiling_mode(False)
317
+ torch._C._set_graph_executor_optimize(False)
318
+
319
  if __name__ == "__main__":
320
  formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
321