Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

make invokeai-model-install use the --root argument correctly

Browse Source
This commit is contained in:
Lincoln Stein 2024-03-16 20:28:49 -04:00 committed by psychedelicious
parent e76cc71e81
commit 8cd65755ef
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/frontend/install/model_install.py
View File

@ -11,6 +11,7 @@ It is currently named model_install2.py, but will ultimately replace model_insta
import argparse import argparse
import curses import curses
import pathlib
import sys import sys
import traceback import traceback
import warnings import warnings
@ -594,18 +595,19 @@ def main() -> None:
parser.add_argument( parser.add_argument(
"--root_dir", "--root_dir",
dest="root", dest="root",
type=str, type=pathlib.Path,
default=None, default=None,
help="path to root of install directory", help="path to root of install directory",
) )
opt = parser.parse_args() opt = parser.parse_args()
invoke_args: dict[str, Any] = {} invoke_args: dict[str, Any] = {}
if opt.root:
invoke_args["root"] = opt.root
if opt.full_precision: if opt.full_precision:
invoke_args["precision"] = "float32" invoke_args["precision"] = "float32"
config.update_config(invoke_args) config.update_config(invoke_args)
if opt.root:
config.set_root(opt.root)
logger = InvokeAILogger().get_logger(config=config) logger = InvokeAILogger().get_logger(config=config)
try: try: