Commit
•
04007bd
1
Parent(s):
ce37a1c
small fixes
Browse files
app.py
CHANGED
|
@@ -316,6 +316,13 @@ with demo:
|
|
| 316 |
)
|
| 317 |
gr.Markdown(description)
|
| 318 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 319 |
if __name__ == "__main__":
|
| 320 |
formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
|
| 321 |
|
|
|
|
| 316 |
)
|
| 317 |
gr.Markdown(description)
|
| 318 |
|
| 319 |
+
torch.set_num_threads(1)
|
| 320 |
+
torch.set_num_interop_threads(1)
|
| 321 |
+
|
| 322 |
+
torch._C._jit_set_profiling_executor(False)
|
| 323 |
+
torch._C._jit_set_profiling_mode(False)
|
| 324 |
+
torch._C._set_graph_executor_optimize(False)
|
| 325 |
+
|
| 326 |
if __name__ == "__main__":
|
| 327 |
formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
|
| 328 |
|