aurelio-ai
/

clip-and-git-images

juanpablomesa commited on Nov 16, 2023

Commit

da91a0c

1 Parent(s): 16f4f18

Modified max length of captions from 50 to 35 tokens

Files changed (1) hide show

handler.py CHANGED Viewed

@@ -207,7 +207,7 @@ class EndpointHandler:
                     git_pixel_values = batch_git.pixel_values.to(self.model_git.device)
                     # get image captions
                     generated_ids = self.model_git.generate(
-                        pixel_values=git_pixel_values, max_length=50
                     )
                     generated_captions = self.processor_git.batch_decode(

                     git_pixel_values = batch_git.pixel_values.to(self.model_git.device)
                     # get image captions
                     generated_ids = self.model_git.generate(
+                        pixel_values=git_pixel_values, max_length=35
                     )
                     generated_captions = self.processor_git.batch_decode(