github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-18 08:12:22 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(config): use correct config in textual_inversion_training

Browse Source
This commit is contained in:
psychedelicious
2024-03-11 23:14:01 +11:00
parent 7716a4a8c7
commit 9539ecce79
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/backend/training/textual_inversion_training.py
View File

@@ -40,6 +40,7 @@ from transformers import CLIPTextModel, CLIPTokenizer
# invokeai stuff # invokeai stuff
from invokeai.app.services.config import InvokeAIAppConfig, PagingArgumentParser from invokeai.app.services.config import InvokeAIAppConfig, PagingArgumentParser
from invokeai.app.services.config.config_default import get_config
from invokeai.backend.install.install_helper import initialize_record_store from invokeai.backend.install.install_helper import initialize_record_store
from invokeai.backend.model_manager import BaseModelType, ModelType from invokeai.backend.model_manager import BaseModelType, ModelType
@@ -77,7 +78,7 @@ def save_progress(text_encoder, placeholder_token_id, accelerator, placeholder_t
def parse_args() -> Namespace: def parse_args() -> Namespace:
config = InvokeAIAppConfig.get_config() config = get_config()
parser = PagingArgumentParser(description="Textual inversion training") parser = PagingArgumentParser(description="Textual inversion training")
general_group = parser.add_argument_group("General") general_group = parser.add_argument_group("General")
model_group = parser.add_argument_group("Models and Paths") model_group = parser.add_argument_group("Models and Paths")
@@ -98,7 +99,7 @@ def parse_args() -> Namespace:
"--root_dir", "--root_dir",
"--root", "--root",
type=Path, type=Path,
default=config.root, default=config.root_path,
help="Path to the invokeai runtime directory", help="Path to the invokeai runtime directory",
) )
general_group.add_argument( general_group.add_argument(
@@ -113,7 +114,7 @@ def parse_args() -> Namespace:
general_group.add_argument( general_group.add_argument(
"--output_dir", "--output_dir",
type=Path, type=Path,
default=f"{config.root}/text-inversion-model", default=f"{config.root_path}/text-inversion-model",
help="The output directory where the model predictions and checkpoints will be written.", help="The output directory where the model predictions and checkpoints will be written.",
) )
model_group.add_argument( model_group.add_argument(
@@ -549,7 +550,7 @@ def do_textual_inversion_training(
# setting up things the way invokeai expects them # setting up things the way invokeai expects them
if not os.path.isabs(output_dir): if not os.path.isabs(output_dir):
output_dir = os.path.join(config.root, output_dir) output_dir = config.root_path / output_dir
logging_dir = output_dir / logging_dir logging_dir = output_dir / logging_dir