Fix wrong name for FuyuProcessorKwargs

huggingface · Aug 14, 2024 · aa3bc0b · aa3bc0b
1 parent a5f7b7b
commit aa3bc0b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/transformers/models/fuyu/processing_fuyu.py b/src/transformers/models/fuyu/processing_fuyu.py
@@ -61,7 +61,7 @@
 BEGINNING_OF_ANSWER_STRING = "<0x04>"  # <boa>
 
 
-class Blip2ProcessorKwargs(ProcessingKwargs, total=False):
+class FuyuProcessorKwargs(ProcessingKwargs, total=False):
     _defaults = {
         "text_kwargs": {
             "add_special_tokens": True,
@@ -486,7 +486,7 @@ def __call__(
         text: Optional[Union[str, List[str], TextInput, PreTokenizedInput]] = None,
         audio=None,
         videos=None,
-        **kwargs: Unpack[Blip2ProcessorKwargs],
+        **kwargs: Unpack[FuyuProcessorKwargs],
     ) -> "FuyuBatchFeature":
         """
         Main method to prepare for the model one or several sequences(s) and image(s). This method forwards the `text`
@@ -532,7 +532,7 @@ def __call__(
             images, text = text, images
 
         output_kwargs = self._merge_kwargs(
-            Blip2ProcessorKwargs,
+            FuyuProcessorKwargs,
             tokenizer_init_kwargs=self.tokenizer.init_kwargs,
             **kwargs,
         )