Spaces:
No application file
No application file
File size: 1,861 Bytes
3883c60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from webui.args import args # Will show help message if needed
import os
# Set custom default huggingface download path
if not args.no_data_cache:
os.environ['XDG_CACHE_HOME'] = os.getenv('XDG_CACHE_HOME', os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'models', 'unclassified'))
os.environ['HF_HOME'] = os.getenv('HF_HOME', os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'models', 'unclassified'))
os.environ['MUSICGEN_ROOT'] = os.getenv('MUSICGEN_ROOT', os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'models', 'musicgen'))
os.environ['HF_HUB_CACHE'] = os.getenv('HF_HUB_CACHE', os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'models', 'hf_cache')) # Experimental, due to some people being unable to install from this variable missing, set a default here.
# Set custom gradio temp dir
os.environ['GRADIO_TEMP_DIR'] = os.getenv('GRADIO_TEMP_DIR', os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'temp'))
from autodebug.prelaunch import prelaunch_checks
from autodebug import autodebug
try:
print('Checking installs and venv + autodebug checks')
prelaunch_checks()
print('Activating extensions')
import webui.extensionlib.extensionmanager as em
for e in em.states.values():
e.activate()
print('Preparing')
from webui.modules.implementations.tts_monkeypatching import patch as patch1
patch1()
# from webui.modules.implementations.gradio_monkeypatching import patch as patch2
# patch2()
#
from webui.modules.implementations.huggingface_hub_monkeypatching import patch as patch3
patch3()
import torch
print('Launching, cuda available:', torch.cuda.is_available())
from webui.webui import launch_webui
launch_webui()
except Exception as e:
autodebug.catcher(e)
|