Internal

PiperOrigin-RevId: 616075000
google-research · Mar 15, 2024 · 3305949 · 3305949
1 parent 9bf9ad2
commit 3305949
Showing 1 changed file with 1 addition and 0 deletions.
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)
+