Update src/pipeline.py

src/pipeline.py

@@ -190,7 +190,6 @@ class FluxSingleTransformerBlock(nn.Module):
         tinfo: Dict[str, Any] = None,
     ):
         if tinfo is not None:
-            print("its working ")
             m_a, m_c, mom, u_a, u_c, u_m = calcular_fusion(hidden_states, tinfo)
         else:
             m_a, m_c, mom, u_a, u_c, u_m = (ghanta.hacer_nada, ghanta.hacer_nada, ghanta.hacer_nada, ghanta.hacer_nada, ghanta.hacer_nada, ghanta.hacer_nada)
@@ -370,7 +369,7 @@ class FluxTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOrig
 
         self.norm_out = AdaLayerNormContinuous(self.inner_dim, self.inner_dim, elementwise_affine=False, eps=1e-6)
         self.proj_out = nn.Linear(self.inner_dim, patch_size * patch_size * self.out_channels, bias=True)
-        ratio: float = 0.5
+        ratio: float = 0.3
         down: int = 1
         sx: int = 4
         sy: int = 4