Rename method (#1)
Browse files- Rename method (8a50b673e39377571e91219275994bcc16073f35)
Co-authored-by: Niels Rogge <[email protected]>
- processing_kosmos2.py +1 -1
processing_kosmos2.py
CHANGED
@@ -336,7 +336,7 @@ class Kosmos2Processor(ProcessorMixin):
|
|
336 |
"""
|
337 |
return self.tokenizer.decode(*args, **kwargs)
|
338 |
|
339 |
-
def
|
340 |
|
341 |
caption = text.split("</image>")[-1]
|
342 |
if cleanup_and_extract:
|
|
|
336 |
"""
|
337 |
return self.tokenizer.decode(*args, **kwargs)
|
338 |
|
339 |
+
def post_process_generation(self, text, cleanup_and_extract=True):
|
340 |
|
341 |
caption = text.split("</image>")[-1]
|
342 |
if cleanup_and_extract:
|