fix: limit CPU usage for VAD onnxruntime inference session by setting OMP_NUM_THREADS
Signed-off-by: makaveli10 <vineet.suryan@collabora.com>
This commit is contained in:
@@ -53,7 +53,13 @@ python3 run_server.py -p 9090 \
|
|||||||
-trt /home/TensorRT-LLM/examples/whisper/whisper_small \
|
-trt /home/TensorRT-LLM/examples/whisper/whisper_small \
|
||||||
-m
|
-m
|
||||||
```
|
```
|
||||||
|
#### Controlling OpenMP Threads
|
||||||
|
To control the number of threads used by OpenMP, you can set the `OMP_NUM_THREADS` environment variable. This is useful for managing CPU resources and ensuring consistent performance. If not specified, `OMP_NUM_THREADS` is set to `1` by default. You can change this by using the `--omp_num_threads` argument:
|
||||||
|
```bash
|
||||||
|
python3 run_server.py --port 9090 \
|
||||||
|
--backend faster_whisper \
|
||||||
|
--omp_num_threads 4
|
||||||
|
```
|
||||||
|
|
||||||
### Running the Client
|
### Running the Client
|
||||||
- Initializing the client:
|
- Initializing the client:
|
||||||
|
|||||||
+9
-1
@@ -1,5 +1,5 @@
|
|||||||
import argparse
|
import argparse
|
||||||
from whisper_live.server import TranscriptionServer
|
import os
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
@@ -21,12 +21,20 @@ if __name__ == "__main__":
|
|||||||
parser.add_argument('--trt_multilingual', '-m',
|
parser.add_argument('--trt_multilingual', '-m',
|
||||||
action="store_true",
|
action="store_true",
|
||||||
help='Boolean only for TensorRT model. True if multilingual.')
|
help='Boolean only for TensorRT model. True if multilingual.')
|
||||||
|
parser.add_argument('--omp_num_threads', '-omp',
|
||||||
|
type=int,
|
||||||
|
default=1,
|
||||||
|
help="Number of threads to use for OpenMP")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
if args.backend == "tensorrt":
|
if args.backend == "tensorrt":
|
||||||
if args.trt_model_path is None:
|
if args.trt_model_path is None:
|
||||||
raise ValueError("Please Provide a valid tensorrt model path")
|
raise ValueError("Please Provide a valid tensorrt model path")
|
||||||
|
|
||||||
|
if "OMP_NUM_THREADS" not in os.environ:
|
||||||
|
os.environ["OMP_NUM_THREADS"] = str(args.omp_num_threads)
|
||||||
|
|
||||||
|
from whisper_live.server import TranscriptionServer
|
||||||
server = TranscriptionServer()
|
server = TranscriptionServer()
|
||||||
server.run(
|
server.run(
|
||||||
"0.0.0.0",
|
"0.0.0.0",
|
||||||
|
|||||||
Reference in New Issue
Block a user