Spaces:
No application file
No application file
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) | |