pyink

Author: hengtaoguo

src/MaxText/decode.py

@@ -106,7 +106,10 @@ def main(argv: Sequence[str]) -> None:
 
     prefill_length -= image_offsets
     text = multimodal_utils.reformat_prompt(
-        text, image_placeholder=config.image_placeholder, model_name=config.model_name, num_images=processor_outputs.num_images
+        text,
+        image_placeholder=config.image_placeholder,
+        model_name=config.model_name,
+        num_images=processor_outputs.num_images,
     )
 
   metadata = engine.get_tokenizer()

src/MaxText/multimodal/preprocessor.py

@@ -42,4 +42,4 @@ def preprocess_mm_data(config):
   else:
     raise ValueError(f"Model {config.model_name} not supported for multimodal preprocessing.")
 
-  return processor_outputs
+  return processor_outputs