How to create multiple image embeddings from a single text embedding

drboog / Shifted_Diffusion

Code for Shifted Diffusion for Text-to-image Generation (CVPR 2023)

Creative Commons Zero v1.0 Universal

159 stars 11 forks source link

How to create multiple image embeddings from a single text embedding #12

Open tenten0727 opened 1 year ago

tenten0727 commented 1 year ago

I wanted to generate a variety of image embeddings for a single text input, but when I ran the code, I found that it generated the same image embeddings for each text input.

To generate a variety of image embeddings from a single text input, I changed line 286 of model_lib/decoder/clip_prior.py to

self.std = self.log_std_init.exp()

Is this collect?

Thank you very much!