update

omini_control/block.py

@@ -98,8 +98,6 @@ def attn_forward(
             cond_key = attn.norm_k(cond_key)
 
     if cond_rotary_emb is not None:
-        print("cond_query.shape", cond_query.shape)
-        print("cond_rotary_emb.shape", cond_rotary_emb[0].shape)
         cond_query = apply_rotary_emb(cond_query, cond_rotary_emb)
         cond_key = apply_rotary_emb(cond_key, cond_rotary_emb)
 

omini_control/condition.py

@@ -93,7 +93,6 @@ class Condition(object):
         cond_img = (
             cond_img - pipe.vae.config.shift_factor
         ) * pipe.vae.config.scaling_factor
-        print("Cond_img.shape", cond_img.shape)
         cond_tokens = pipe._pack_latents(cond_img, *cond_img.shape)
         cond_ids = pipe._prepare_latent_image_ids(
             cond_img.shape[0],
@@ -102,8 +101,6 @@ class Condition(object):
             pipe.device,
             pipe.dtype,
         )
-        print("Cond_tokens.shape", cond_tokens.shape)
-        print("Cond_ids.shape", cond_ids.shape)
         return cond_tokens, cond_ids
 
     def encode(self, pipe: FluxPipeline) -> Tuple[torch.Tensor, torch.Tensor, int]: