ydshieh
/

kosmos-2-patch14-224

Image-Text-to-Text

Inference Endpoints

Model card Files Files and versions Community

ydshieh HF staff commited on Aug 19, 2023

Commit

f7f3e17

•

1 Parent(s): 83726ae

Upload processing_kosmos2.py

Files changed (1) hide show

processing_kosmos2.py +1 -1

processing_kosmos2.py CHANGED Viewed

@@ -59,7 +59,7 @@ class Kosmos2Processor(ProcessorMixin):
     """
     attributes = ["image_processor", "tokenizer"]
     image_processor_class = "CLIPImageProcessor"
-    tokenizer_class = ("AutoTokenizer",)
     def __init__(self, image_processor, tokenizer):
         tokenizer.return_token_type_ids = False

     """
     attributes = ["image_processor", "tokenizer"]
     image_processor_class = "CLIPImageProcessor"
+    tokenizer_class = "AutoTokenizer"
     def __init__(self, image_processor, tokenizer):
         tokenizer.return_token_type_ids = False