travel back

controlnet_flux.py

@@ -45,10 +45,10 @@ class FluxControlNetModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
         self,
         patch_size: int = 1,
         in_channels: int = 64,
-          num_layers: int = 24,
-        num_single_layers: int = 48,
-        attention_head_dim: int = 256,
-        num_attention_heads: int = 35,
+          num_layers: int = 19,
+        num_single_layers: int = 38,
+        attention_head_dim: int = 128,
+        num_attention_heads: int = 24,
         joint_attention_dim: int = 4096,
         pooled_projection_dim: int = 768,
         guidance_embeds: bool = False,

transformer_flux.py

@@ -278,10 +278,10 @@ class FluxTransformer2DModel(
         self,
         patch_size: int = 1,
         in_channels: int = 64,
-        num_layers: int = 24,
-        num_single_layers: int = 48,
-        attention_head_dim: int = 256,
-        num_attention_heads: int = 32,
+        num_layers: int = 19,
+        num_single_layers: int = 38,
+        attention_head_dim: int = 128,
+        num_attention_heads: int = 24,
         joint_attention_dim: int = 4096,
         pooled_projection_dim: int = 768,
         guidance_embeds: bool = False,