mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
'''
|
|
ldm.invoke.globals defines a small number of global variables that would
|
|
otherwise have to be passed through long and complex call chains.
|
|
|
|
It defines a Namespace object named "Globals" that contains
|
|
the attributes:
|
|
|
|
- root - the root directory under which "models" and "outputs" can be found
|
|
- initfile - path to the initialization file
|
|
- try_patchmatch - option to globally disable loading of 'patchmatch' module
|
|
- always_use_cpu - force use of CPU even if GPU is available
|
|
'''
|
|
|
|
import os
|
|
import os.path as osp
|
|
from argparse import Namespace
|
|
|
|
Globals = Namespace()
|
|
|
|
# This is usually overwritten by the command line and/or environment variables
|
|
if os.environ.get('INVOKEAI_ROOT'):
|
|
Globals.root = osp.abspath(os.environ.get('INVOKEAI_ROOT'))
|
|
elif os.environ.get('VIRTUAL_ENV'):
|
|
Globals.root = osp.abspath(osp.join(os.environ.get('VIRTUAL_ENV'), '..'))
|
|
else:
|
|
Globals.root = osp.abspath(osp.expanduser('~/invokeai'))
|
|
|
|
# Where to look for the initialization file
|
|
Globals.initfile = 'invokeai.init'
|
|
|
|
# Awkward workaround to disable attempted loading of pypatchmatch
|
|
# which is causing CI tests to error out.
|
|
Globals.try_patchmatch = True
|
|
|
|
# Use CPU even if GPU is available (main use case is for debugging MPS issues)
|
|
Globals.always_use_cpu = False
|