ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added ability to use civit ai url ar model name and built a model downloader and cache manager for it

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-13 13:09:51 -06:00
parent 1487d13191
commit 355c80df07
4 changed files with 236 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
toolkit/paths.py
View File

@@ -5,6 +5,12 @@ 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")
# check if ENV variable is set
if 'MODELS_PATH' in os.environ:
MODELS_PATH = os.environ['MODELS_PATH']
else:
MODELS_PATH = os.path.join(TOOLKIT_ROOT, "models")
def get_path(path):
# we allow absolute paths, but if it is not absolute, we assume it is relative to the toolkit root