Fix unbounded variables issue

huggingface · Jul 26, 2024 · 116ea57 · 116ea57
1 parent 469f825
commit 116ea57
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/transformers/models/idefics/processing_idefics.py b/src/transformers/models/idefics/processing_idefics.py
@@ -207,7 +207,7 @@ def __call__(
         padding: Union[bool, str, PaddingStrategy] = False,
         truncation: Union[bool, str, TruncationStrategy] = None,
         max_length: Optional[int] = None,
-        return_tensors: Optional[Union[str, TensorType]] = None,
+        return_tensors: Optional[Union[str, TensorType]] = "pt",
         legacy=True,
         prompts: Optional[Union[List[TextInput], List[List[TextInput]]]] = None,
         transform: Callable = None,

diff --git a/src/transformers/pipelines/base.py b/src/transformers/pipelines/base.py
@@ -1204,6 +1204,8 @@ def get_iterator(
             feature_extractor = self.image_processor
         elif self.processor is not None:
             feature_extractor = self.processor
+        else:
+            feature_extractor = None
         collate_fn = no_collate_fn if batch_size == 1 else pad_collate_fn(self.tokenizer, feature_extractor)
         dataloader = DataLoader(dataset, num_workers=num_workers, batch_size=batch_size, collate_fn=collate_fn)
         model_iterator = PipelineIterator(dataloader, self.forward, forward_params, loader_batch_size=batch_size)

diff --git a/src/transformers/pipelines/image_text_to_text.py b/src/transformers/pipelines/image_text_to_text.py
@@ -75,7 +75,7 @@ def count_images_in_chat(chat):
 @add_end_docstrings(build_pipeline_init_args(has_processor=True))
 class ImageTextToTextPipeline(Pipeline):
     """
-    Image-text-to-text pipeline. This pipeline generates text given an image and text.
+    Image-text-to-text pipeline using an `AutoModelForImageTextToText`. This pipeline generates text given an image and text.
 
     Example: