Could you implement Omnigen Training script to fine-tune? thks

[ivaxsirc]

OmniGen is a unified image generation model that you can use to perform various tasks, including but not limited to text-to-image generation, subject-driven generation, Identity-Preserving Generation, image editing, and image-conditioned generation. OmniGen doesn't need additional plugins or operations, it can automatically identify the features (e.g., required object, human pose, depth mapping) in input images according to the text prompt.

And provide a training script train.py to fine-tune.

https://github.com/VectorSpaceLab/OmniGen