File size: 286 Bytes
87f70d4
 
 
 
 
 
 
1
2
3
4
5
6
7
8
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)