Adding the missing new line

Author: parambole

MaxText/layers/blocks.py

@@ -699,4 +699,5 @@ def __call__(self, input_images, deterministic=False):
     if len(self.vision_encoder_layer) > 1:
       # vision embedder / projection layer, not frozen in most cases, trained / finetuned together with main model
       embeddings = self.vision_encoder_layer[1](config=cfg, mesh=mesh)(embeddings)
-    return embeddings
+    return embeddings
+  

MaxText/layers/models.py

@@ -156,4 +156,5 @@ def __call__(
           model_mode=model_mode,
       )
 
-    return logits
+    return logits
+