From 6a8e76e37f1d3d4aa09403b118025e5b2ffc7603 Mon Sep 17 00:00:00 2001 From: dogewanwan <66496316+dogewanwan@users.noreply.github.com> Date: Wed, 24 Aug 2022 21:46:50 +0300 Subject: [PATCH] Check for the embedding manager --- webui.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/webui.py b/webui.py index 33da6c10e..55654f559 100644 --- a/webui.py +++ b/webui.py @@ -469,8 +469,9 @@ def process_images(outpath, func_init, func_sample, prompt, seed, sampler_name, def load_embeddings(fp): + if fp is not None and hasattr(model, "embedding_manager"): # load the file - model.embedding_manager.load(fp.name) + model.embedding_manager.load(fp.name) def txt2img(prompt: str, ddim_steps: int, sampler_name: str, use_GFPGAN: bool, prompt_matrix: bool, ddim_eta: float, n_iter: int, batch_size: int, cfg_scale: float, seed: int, height: int, width: int, embeddings_fp):