From 3305949b4069bfcc05f23980e64b809746eaab24 Mon Sep 17 00:00:00 2001 From: Anurag Arnab Date: Fri, 15 Mar 2024 03:39:05 -0700 Subject: [PATCH] Internal PiperOrigin-RevId: 616075000 --- scenic/projects/baselines/segment_anything/modeling/sam.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scenic/projects/baselines/segment_anything/modeling/sam.py b/scenic/projects/baselines/segment_anything/modeling/sam.py index b24316f4..96049520 100644 --- 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) +