michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Check for the embedding manager

Browse source
This commit is contained in:
dogewanwan 2022-08-24 21:46:50 +03:00 committed by GitHub
parent 781f054a20
commit 6a8e76e37f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
webui.py
View file

@ -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):