clean

Author: hengtaoguo

src/maxtext/layers/attentions.py

@@ -789,7 +789,6 @@ def init_rotary_embedding(self):
     rope_use_scale = self.config.rope_use_scale
     if self.is_vision:
       if self.config.model_name.startswith("qwen3-omni") or self.config.model_name.startswith("qwen3.5"):
-        print(f"DEBUG: Instantiating Qwen3OmniMoeVisionRotaryEmbedding for {self.config.model_name}")
         rotary_embedding = Qwen3OmniMoeVisionRotaryEmbedding(
             hidden_size=self.config.hidden_size_for_vit,
             num_attention_heads=self.config.num_attention_heads_for_vit,
@@ -813,7 +812,6 @@ def init_rotary_embedding(self):
         raise ValueError(f"Unsupported model type for vision rotary embedding: {self.config.model_name}")
 
     elif self.use_mrope:
-      print(f"DEBUG: Instantiating Qwen3OmniMoeThinkerTextRotaryEmbedding for {self.config.model_name}")
       rotary_embedding = Qwen3OmniMoeThinkerTextRotaryEmbedding(
           min_timescale=self.config.rope_min_timescale,
           max_timescale=self.rope_max_timescale,
@@ -853,7 +851,6 @@ def init_rotary_embedding(self):
           rngs=self.rngs,
       )
     elif self.is_qwen3_hybrid:
-      print(f"DEBUG: Instantiating PartialRotaryEmbedding (hybrid) for {self.config.model_name}")
       rotary_embedding = PartialRotaryEmbedding(
           min_timescale=self.config.rope_min_timescale,
           max_timescale=self.rope_max_timescale,