adjust copybara rules for multimodal refactor (#3101)

PiperOrigin-RevId: 866171684

Co-authored-by: maxtext authors <google-ml-automation@google.com>

Author: copybara-service[bot]

benchmarks/api_server/maxtext_generator.py

@@ -35,8 +35,8 @@
 from dataclasses import dataclass, field
 
 from MaxText import maxengine, pyconfig
-from MaxText.multimodal import processor as mm_processor
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import processor as mm_processor
+from maxtext.multimodal import utils as mm_utils
 from maxtext.utils import max_logging, max_utils
 
 # Set TF log level to avoid verbose startup messages.

src/MaxText/input_pipeline/_input_pipeline_utils.py

@@ -24,8 +24,8 @@
 import numpy as np
 import tensorflow as tf
 from MaxText import tokenizer
-from MaxText.multimodal import processor as mm_processor
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import processor as mm_processor
+from maxtext.multimodal import utils as mm_utils
 from maxtext.utils import max_logging
 
 Features = dict[str, tf.Tensor]
@@ -776,7 +776,7 @@ def map(self, element: dict[str, np.ndarray]) -> dict[str, np.ndarray]:
     second_per_grids = element.get("second_per_grids")
 
     # Call the standalone get_rope_index function from multimodal_utils
-    from MaxText.multimodal import processor_qwen3_omni  # pylint: disable=import-outside-toplevel
+    from maxtext.multimodal import processor_qwen3_omni  # pylint: disable=import-outside-toplevel
 
     # TODO(jfacevedo/hengtaoguo): Now get_rope_index is Qwen3-Omni specific. We should generalize it for other models
     position_ids, mrope_position_deltas = processor_qwen3_omni.get_rope_index(

src/MaxText/layers/decoders.py

@@ -56,7 +56,7 @@
     olmo3,
 )
 from maxtext.inference import page_manager
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import utils as mm_utils
 from maxtext.utils import max_logging
 from maxtext.utils import max_utils
 from maxtext.utils import maxtext_utils

src/MaxText/layers/models.py

@@ -33,7 +33,7 @@
 from MaxText.layers.quantizations import AqtQuantization as Quant
 from MaxText.layers.multi_token_prediction import multi_token_prediction_block_as_linen
 from maxtext.inference import page_manager
-from MaxText.multimodal import processor as mm_processor
+from maxtext.multimodal import processor as mm_processor
 from maxtext.utils import max_utils
 
 # ------------------------------------------------------------------------------

src/MaxText/maxengine.py

@@ -42,7 +42,7 @@
 from MaxText.layers import models, quantizations
 from maxtext.inference import inference_utils
 from maxtext.inference.page_manager import PageManager, PageState
-from MaxText.multimodal import processor as mm_processor
+from maxtext.multimodal import processor as mm_processor
 from maxtext.utils import lora_utils
 from maxtext.utils import max_utils
 from maxtext.utils import maxtext_utils

src/maxtext/decode.py

@@ -26,8 +26,8 @@
 from MaxText import pyconfig
 from maxtext.common import profiler
 from maxtext.common.gcloud_stub import jetstream, is_decoupled
-from MaxText.multimodal import processor as mm_processor
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import processor as mm_processor
+from maxtext.multimodal import utils as mm_utils
 from maxtext.utils import max_utils
 
 _config_lib, engine_api, _token_utils, _tokenizer_api, _token_params_ns = jetstream()
@@ -141,7 +141,7 @@ def main(argv: Sequence[str]) -> None:
     true_length += image_offsets
 
     if config.use_mrope:
-      from MaxText.multimodal import processor_qwen3_omni  # pylint: disable=import-outside-toplevel
+      from maxtext.multimodal import processor_qwen3_omni  # pylint: disable=import-outside-toplevel
 
       position_ids, mrope_position_deltas = processor_qwen3_omni.get_rope_index(
           input_ids=tokens,

src/maxtext/multimodal/processor.py

@@ -14,7 +14,7 @@
 
 """Multimodal data preprocessor router."""
 
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import utils as mm_utils
 
 
 def preprocess_mm_data(config):

src/maxtext/multimodal/processor_gemma3.py

@@ -19,7 +19,7 @@
 import numpy as np
 from PIL import Image
 
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import utils as mm_utils
 
 # Constants for Gemma3-specific processing
 GEMMA_DEFAULT_IMAGE_SIZE = 896

src/maxtext/multimodal/processor_llama4.py

@@ -21,7 +21,7 @@
 import numpy as np
 from PIL import Image
 
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import utils as mm_utils
 
 # Constants for Llama4-specific processing
 LLAMA4_TILE_SIZE = 336

src/maxtext/multimodal/processor_qwen3_omni.py

@@ -31,7 +31,7 @@
 except ImportError:
   decord = None
 
-from MaxText.multimodal import utils as mm_utils
+from maxtext.multimodal import utils as mm_utils
 from maxtext.utils import max_logging
 
 # Image constants.