Update modules/aesthetic_clip.py
Co-authored-by: Víctor Gallego <vicgalle@ucm.es>
This commit is contained in:
parent
f7df06a981
commit
9b7705e057
|
@ -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",
|
||||
|
|
Loading…
Reference in New Issue