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

Update modules/aesthetic_clip.py

Browse Source 
Co-authored-by: Víctor Gallego <vicgalle@ucm.es>
This commit is contained in:
MalumaDev 2022-10-15 18:40:34 +02:00 committed by GitHub
parent f7df06a981
commit 9b7705e057
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/aesthetic_clip.py
View File

@ -70,7 +70,7 @@ def generate_imgs_embd(name, folder, batch_size):
torch.cuda.empty_cache() torch.cuda.empty_cache()
res = f""" res = f"""
Done generating embedding for {name}! Done generating embedding for {name}!
Hypernetwork saved to {html.escape(path)} Aesthetic embedding saved to {html.escape(path)}
""" """
shared.update_aesthetic_embeddings() shared.update_aesthetic_embeddings()
return gr.Dropdown(sorted(aesthetic_embeddings.keys()), label="Imgs embedding", return gr.Dropdown(sorted(aesthetic_embeddings.keys()), label="Imgs embedding",