This repository contains the model described in OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows.

Code: https://github.com/jacklishufan/OmniFlows

Usage:

from omniflow import  OmniFlowPipeline

pipeline = OmniFlowPipeline.load_pretrained('ckpts/v0.5',device='cuda')

pipeline.cfg_mode = 'new'
imgs = pipeline("portrait of a cyberpunk girl with neon tattoos and a visor,staring intensely. Standing on top of a building",height=512,width=512,add_token_embed=0,task='t2i')

See Notebook for more examples