StyleCLIP icon indicating copy to clipboard operation
StyleCLIP copied to clipboard

[Feature Request] Save model checkpoints & upload starting images

Open Christoph-Schuhmann opened this issue 4 years ago • 2 comments

WOW...! EXTREMELY COOL!!! :)

Could you make a version of the colab where you can save model checkpoints and where you can upload an image as a starting point ? :)

Christoph-Schuhmann avatar Feb 19 '21 01:02 Christoph-Schuhmann

Hi @Christoph-Schuhmann ,

I am glad that you are excited about the project! 🤩

What exactly do you mean by saving checkpoint? Both StyleGAN and CLIP remain with their initial weights along the optimization. Do you mean maybe to save the optimized latent codes? If so, would you like to save them along the optimization or just the final result?

To edit your own image you can upload a latent code, and use the latent_path argument. You can refer to this issue for additional details. Let me know if it solves your problem. It might be useful to integrate the relevant code from pSp's notebook into SyleCLIP's notebook, I will do it soon 🙂

orpatashnik avatar Feb 19 '21 08:02 orpatashnik

Ah... I was assuming that the StyleGAN weights were finetuned. :P

So it would be great to be able to export the latent values via the interface to a .pt :)

And an interface for inversion would also be AWESOME! :)

Christoph-Schuhmann avatar Feb 19 '21 10:02 Christoph-Schuhmann