VectorSpaceLab / OmniGen

OmniGen: Unified Image Generation. https://arxiv.org/pdf/2409.11340
MIT License
2.57k stars 190 forks source link

Open weights or frozen? #57

Closed DarkAlchy closed 2 weeks ago

DarkAlchy commented 2 weeks ago

I see no mention of this, but it reads as if it is not frozen.

staoxiao commented 2 weeks ago

During training, we only freeze the VAE and keep other parameters trainable.

DarkAlchy commented 2 weeks ago

Thank you.