Files
ai-toolkit/toolkit/paths.py
2023-07-23 11:24:12 -06:00

14 lines
489 B
Python

import os
TOOLKIT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
CONFIG_ROOT = os.path.join(TOOLKIT_ROOT, 'config')
SD_SCRIPTS_ROOT = os.path.join(TOOLKIT_ROOT, "repositories", "sd-scripts")
REPOS_ROOT = os.path.join(TOOLKIT_ROOT, "repositories")
def get_path(path):
# we allow absolute paths, but if it is not absolute, we assume it is relative to the toolkit root
if not os.path.isabs(path):
path = os.path.join(TOOLKIT_ROOT, path)
return path