From 5ee5c5a012a4a25fe214767d29489d2ae99f9097 Mon Sep 17 00:00:00 2001 From: Eugene Brodsky Date: Sat, 28 Jan 2023 19:09:16 -0500 Subject: [PATCH] (training) correctly import TI module; fix type annotation --- ldm/invoke/training/textual_inversion.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ldm/invoke/training/textual_inversion.py b/ldm/invoke/training/textual_inversion.py index 715752472e..835ad2d04e 100755 --- a/ldm/invoke/training/textual_inversion.py +++ b/ldm/invoke/training/textual_inversion.py @@ -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