restore behavior of !import_model; fix initial models bug

This commit is contained in:
Lincoln Stein 2023-02-28 00:45:56 -05:00
parent 3b921cf393
commit 1d77581d96
3 changed files with 29 additions and 13 deletions

View File

@ -747,6 +747,7 @@ class ModelManager(object):
)
return
if convert:
diffuser_path = Path(
Globals.root, "models", Globals.converted_ckpts_dir, model_path.stem
)
@ -759,6 +760,21 @@ class ModelManager(object):
original_config_file=model_config_file,
commit_to_conf=commit_to_conf,
)
else:
model_name = self.import_ckpt_model(
model_path,
config=model_config_file,
model_name=model_name,
model_description=description,
vae=str(
Path(
Globals.root,
"models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt",
)
),
commit_to_conf=commit_to_conf,
)
if commit_to_conf:
self.commit(commit_to_conf)
return model_name

View File

@ -18,7 +18,7 @@ from tqdm import tqdm
from typing import List
import invokeai.configs as configs
from ..generator.diffusers_pipeline import StableDiffusionGeneratorPipeline
from invokeai.generator.diffusers_pipeline import StableDiffusionGeneratorPipeline
from ..globals import Globals, global_cache_dir, global_config_dir
from invokeai.models import ModelManager