from .pipeline_dit import DiTPipeline