Unverified Commit 9b7705e0 authored by MalumaDev's avatar MalumaDev Committed by GitHub

Update modules/aesthetic_clip.py

Co-authored-by: 's avatarVíctor Gallego <vicgalle@ucm.es>
parent f7df06a9
...@@ -70,7 +70,7 @@ def generate_imgs_embd(name, folder, batch_size): ...@@ -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",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment