jy0205 / Pyramid-Flow

Code of Pyramidal Flow Matching for Efficient Video Generative Modeling
https://pyramid-flow.github.io/
MIT License
2.4k stars 233 forks source link

training / finetune code? #117

Closed hrcheng98 closed 2 weeks ago

hrcheng98 commented 1 month ago

Do you have plans to release the training and fine-tuning code in the future? Thanks.❤️

jy0205 commented 1 month ago

The training (fine-tuning) code will be released next week. By the way, we recently switched to flux architecture and retrained the model from scratch to solve the human structure and face distortion problems due to the sd3 weights. We also plan to release the new checkpoint next week (first 384p version and then 768p version after finishing training). Stay tuned!

andrearama commented 3 weeks ago

Any news on when the training and flux models will be release?

Thanks for the great work!

jy0205 commented 3 weeks ago

Soon, we're almost done

feifeiobama commented 3 weeks ago

We have open-sourced the training code and flux models, please feel free to use them and report any issues :D