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

facea15

•

1 Parent(s): 3795303

Update 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 = ("Kosmos2Tokenizer", "Kosmos2TokenizerFast")
     def __init__(self, image_processor, tokenizer):
         tokenizer.return_token_type_ids = False

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