f this paper can be found here: facebookresearch/dit.

Available Pipelines:
	
Pipeline
Tasks
Colab
pipeline_dit.py
Conditional Image Generation
-

Usage example
	


	Copied
from diffusers import DiTPipeline, DPMSolverMultistepScheduler
import torch

pipe = DiTPipeline.from_pretrained("facebook/DiT-XL-2-256", torch_dtype=torch.float16)
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
pipe = pipe.to("cuda")

# pick 