How to use lijiang/Omni-Diffusion with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("lijiang/Omni-Diffusion", trust_remote_code=True, dtype="auto")