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

(training) correctly import TI module; fix type annotation

Browse Source
This commit is contained in:
Eugene Brodsky 2023-01-28 19:09:16 -05:00
parent 3075c99ed2
commit 5ee5c5a012
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/invoke/training/textual_inversion.py
View File

@ -14,13 +14,13 @@ import sys
import traceback
from argparse import Namespace
from pathlib import Path
from typing import List
from typing import List, Tuple
import npyscreen
from omegaconf import OmegaConf
from ldm.invoke.globals import Globals, global_set_root
from ldm.invoke.textual_inversion_training import (
from ldm.invoke.training.textual_inversion_training import (
do_textual_inversion_training,
parse_args,
)
@ -283,7 +283,7 @@ class textualInversionForm(npyscreen.FormMultiPageAction):
else:
return True
def get_model_names(self) -> (List[str], int):
def get_model_names(self) -> Tuple[List[str], int]:
conf = OmegaConf.load(os.path.join(Globals.root, "configs/models.yaml"))
model_names = [
idx