Skip to content

GitLab

Unverified Commit f46abfe4 authored by Valeriy Sofin's avatar Valeriy Sofin Committed by GitHub
Browse files

fixed axes_dims_rope init (huggingface#11641) (#11678)

parent 73a9d585
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/transformers/transformer_flux.py
View file @ f46abfe4
...@@ -241,7 +241,7 @@ class FluxTransformer2DModel( ...@@ -241,7 +241,7 @@ class FluxTransformer2DModel(
joint_attention_dim: int = 4096, joint_attention_dim: int = 4096,
pooled_projection_dim: int = 768, pooled_projection_dim: int = 768,
guidance_embeds: bool = False, guidance_embeds: bool = False,
axes_dims_rope: Tuple[int] = (16, 56, 56), axes_dims_rope: Tuple[int, int, int] = (16, 56, 56),
): ):
super().__init__() super().__init__()
self.out_channels = out_channels or in_channels self.out_channels = out_channels or in_channels
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment