Bark-Voice-Cloning2 / bark /settings.py
aeuhhh's picture
Duplicate from kevinwang676/Bark-Voice-Cloning
77bbad1
raw
history blame contribute delete
No virus
286 Bytes
import os
def initenv(args):
os.environ['SUNO_USE_SMALL_MODELS'] = str("-smallmodels" in args)
os.environ['BARK_FORCE_CPU'] = str("-forcecpu" in args)
os.environ['SUNO_ENABLE_MPS'] = str("-enablemps" in args)
os.environ['SUNO_OFFLOAD_CPU'] = str("-offloadcpu" in args)