AudioGPT / text_to_speech /utils /commons /single_thread_env.py
Datasculptor's picture
Duplicate from AIGC-Audio/AudioGPT
98f685a
raw
history blame
133 Bytes
import os
os.environ["OMP_NUM_THREADS"] = "1"
os.environ['TF_NUM_INTEROP_THREADS'] = '1'
os.environ['TF_NUM_INTRAOP_THREADS'] = '1'