Internal.

anuragarnab · Scenic Authors · commit d5015827f6d8 · 2024-03-15T10:14:48.000-07:00
PiperOrigin-RevId: 616081718
diff --git a/scenic/projects/baselines/segment_anything/modeling/sam.py b/scenic/projects/baselines/segment_anything/modeling/sam.py
@@ -371,3 +371,4 @@ def generate(
   def batch_generate(self, image, padding_mask, upsample_mask=True):
     return jax.vmap(lambda x, y: self.generate(x, y, upsample_mask))(
         image, padding_mask)
+
