Fix OpenBLAS warning (#583)

Fixes the error msg: "OpenBLAS warning: precompiled NUM_THREADS exceeded, adding auxiliary array for thread metadata"
2023-06-20 00:17:18 -07:00 · 2023-06-20 00:17:18 -07:00 · cdbb76cb6f
parent 413c0d285d
commit cdbb76cb6f
1 changed files with 1 additions and 0 deletions
--- a/infer-web.py
+++ b/infer-web.py
@ -10,6 +10,7 @@ import warnings
 import numpy as np
 import torch

+os.environ['OPENBLAS_NUM_THREADS'] = '1'
 os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
 import logging
 import threading