Update modules/aesthetic_clip.py

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

View file

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