Skip to content

Commit 4a62cc5

Browse files
sayakpaulstevhliu
andauthored
Update src/diffusers/pipelines/dit/pipeline_dit.py
Co-authored-by: Steven Liu <[email protected]>
1 parent c96ad66 commit 4a62cc5

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/diffusers/pipelines/dit/pipeline_dit.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,8 @@ class DiTPipeline(DiffusionPipeline):
4646
4747
Parameters:
4848
transformer ([`DiTTransformer2DModel`]):
49-
A class conditioned `DiTTransformer2DModel` to denoise the encoded image latents. It was initially
50-
published as `Transformer2DModel` which is why [the
51-
configuration](https://huggingface.co/facebook/DiT-XL-2-256/blob/main/transformer/config.json#L2) still
52-
shows the class name as `Transformer2DModel`. This mismatch can be safely ignored.
49+
A class conditioned `DiTTransformer2DModel` to denoise the encoded image latents. Initially
50+
published as [Transformer2DModel](https://huggingface.co/facebook/DiT-XL-2-256/blob/main/transformer/config.json#L2) in the config, but the mismatch can be ignored.
5351
vae ([`AutoencoderKL`]):
5452
Variational Auto-Encoder (VAE) model to encode and decode images to and from latent representations.
5553
scheduler ([`DDIMScheduler`]):

0 commit comments

Comments
 (0)