You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: src/diffusers/pipelines/dit/pipeline_dit.py
+2-4Lines changed: 2 additions & 4 deletions
Original file line number
Diff line number
Diff line change
@@ -46,10 +46,8 @@ class DiTPipeline(DiffusionPipeline):
46
46
47
47
Parameters:
48
48
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.
53
51
vae ([`AutoencoderKL`]):
54
52
Variational Auto-Encoder (VAE) model to encode and decode images to and from latent representations.
0 commit comments