Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add models path

Browse Source
This commit is contained in:
Vladimir Mandic
2023-04-20 17:18:31 -04:00
parent 6556ab66c5
commit df424d6d51
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/paths_internal.py
+3 -2
View File
@@ -12,9 +12,10 @@ default_sd_model_file = sd_model_file
# Parse the --data-dir flag first so we can use it as a base for our other argument default values
parser_pre = argparse.ArgumentParser(add_help=False)
parser_pre.add_argument("--data-dir", type=str, default=os.path.dirname(os.path.dirname(os.path.realpath(__file__))), help="base path where all user data is stored",)
parser_pre.add_argument("--models-dir", type=str, default="models", help="base path where all models are stored",)
cmd_opts_pre = parser_pre.parse_known_args()[0]
data_path = cmd_opts_pre.data_dir
models_path = os.path.join(data_path, "models")
models_path = os.path.join(cmd_opts_pre.models_dir)
extensions_dir = os.path.join(data_path, "extensions")
extensions_builtin_dir = os.path.join(script_path, "extensions-builtin")
extensions_builtin_dir = os.path.join(data_path, "extensions-builtin")
requirements.txt
+1 -1
View File
@@ -48,8 +48,8 @@ torchvision
tqdm
voluptuous
yapf
scikit-image
scikit-image==0.19.3
accelerate==0.18.0
opencv-python==4.7.0.72
diffusers==0.15.0