





from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_sentencepiece_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


_lowercase		:     int 					={
    """configuration_marian""": ["""MARIAN_PRETRAINED_CONFIG_ARCHIVE_MAP""", """MarianConfig""", """MarianOnnxConfig"""],
}

try:
		if not is_sentencepiece_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase		:     Any 					=["""MarianTokenizer"""]

try:
		if not is_torch_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase		:     Any 					=[
		    """MARIAN_PRETRAINED_MODEL_ARCHIVE_LIST""",
		    """MarianForCausalLM""",
		    """MarianModel""",
		    """MarianMTModel""",
		    """MarianPreTrainedModel""",
		]

try:
		if not is_tf_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase		:     Dict 					=["""TFMarianModel""", """TFMarianMTModel""", """TFMarianPreTrainedModel"""]

try:
		if not is_flax_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase		:     Optional[Any] 					=["""FlaxMarianModel""", """FlaxMarianMTModel""", """FlaxMarianPreTrainedModel"""]

if TYPE_CHECKING:
		from .configuration_marian import MARIAN_PRETRAINED_CONFIG_ARCHIVE_MAP, MarianConfig, MarianOnnxConfig

		try:
				if not is_sentencepiece_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .tokenization_marian import MarianTokenizer

		try:
				if not is_torch_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_marian import (
				    MARIAN_PRETRAINED_MODEL_ARCHIVE_LIST,
				    MarianForCausalLM,
				    MarianModel,
				    MarianMTModel,
				    MarianPreTrainedModel,
				)

		try:
				if not is_tf_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_tf_marian import TFMarianModel, TFMarianMTModel, TFMarianPreTrainedModel

		try:
				if not is_flax_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_flax_marian import FlaxMarianModel, FlaxMarianMTModel, FlaxMarianPreTrainedModel

else:
		import sys

		_lowercase		:     List[str] 					=_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)







import unittest

import requests

from transformers import MODEL_FOR_VISION_2_SEQ_MAPPING, TF_MODEL_FOR_VISION_2_SEQ_MAPPING, is_vision_available
from transformers.pipelines import pipeline
from transformers.testing_utils import (
    is_pipeline_test,
    is_torch_available,
    require_tf,
    require_torch,
    require_vision,
    slow,
)

from .test_pipelines_common import ANY


if is_torch_available():
		from transformers.pytorch_utils import is_torch_greater_or_equal_than_1_11
else:
		_lowercase		:     Dict 					=False


if is_vision_available():
		from PIL import Image
else:






		class 	UpperCamelCase_     :



							@staticmethod
							def      __a    (		*lowerCamelCase    :    Any					,			**lowerCamelCase    :    List[Any]	):
									pass






@is_pipeline_test
@require_vision
class 	UpperCamelCase_     (     unittest.TestCase       ):
					_a			:	int							=				MODEL_FOR_VISION_2_SEQ_MAPPING
					_a			:	int							=				TF_MODEL_FOR_VISION_2_SEQ_MAPPING



					def      __a    (		self    :    int					,			lowerCamelCase    :    Union[str, Any]					,			lowerCamelCase    :    Optional[int]					,			lowerCamelCase    :    Union[str, Any]	):
							lowerCamelCase_  :    List[str]   	=			pipeline('image-to-text'					,			model=lowerCamelCase					,			tokenizer=lowerCamelCase					,			image_processor=lowerCamelCase	)
							lowerCamelCase_  :    str   	=			[
							    Image.open('./tests/fixtures/tests_samples/COCO/000000039769.png'	),
							    './tests/fixtures/tests_samples/COCO/000000039769.png',
							]
							return pipe, examples



					def      __a    (		self    :    List[Any]					,			lowerCamelCase    :    Any					,			lowerCamelCase    :    List[str]	):
							lowerCamelCase_  :    str   	=			pipe(lowerCamelCase	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        [{'generated_text': ANY(lowerCamelCase	)}],
							        [{'generated_text': ANY(lowerCamelCase	)}],
							    ]					,			)



					@require_tf
					def      __a    (		self    :    Union[str, Any]	):
							lowerCamelCase_  :    Tuple   	=			pipeline('image-to-text'					,			model='hf-internal-testing/tiny-random-vit-gpt2'					,			framework='tf'	)
							lowerCamelCase_  :    Optional[int]   	=			'./tests/fixtures/tests_samples/COCO/000000039769.png'

							lowerCamelCase_  :    Optional[int]   	=			pipe(lowerCamelCase	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        {
							            'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							        },
							    ]					,			)

							lowerCamelCase_  :    List[Any]   	=			pipe([image, image]	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        [
							            {
							                'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							            }
							        ],
							        [
							            {
							                'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							            }
							        ],
							    ]					,			)

							lowerCamelCase_  :    str   	=			pipe(lowerCamelCase					,			max_new_tokens=1	)
							self.assertEqual(
							    lowerCamelCase					,			[{'generated_text': 'growth'}]					,			)



					@require_torch
					def      __a    (		self    :    Tuple	):
							lowerCamelCase_  :    Union[str, Any]   	=			pipeline('image-to-text'					,			model='hf-internal-testing/tiny-random-vit-gpt2'	)
							lowerCamelCase_  :    Any   	=			'./tests/fixtures/tests_samples/COCO/000000039769.png'

							lowerCamelCase_  :    int   	=			pipe(lowerCamelCase	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        {
							            'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							        },
							    ]					,			)

							lowerCamelCase_  :    Tuple   	=			pipe([image, image]	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        [
							            {
							                'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							            }
							        ],
							        [
							            {
							                'generated_text': 'growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO'
							            }
							        ],
							    ]					,			)



					@require_torch
					def      __a    (		self    :    Dict	):
							lowerCamelCase_  :    Any   	=			pipeline('image-to-text'					,			model='hf-internal-testing/tiny-random-BlipForConditionalGeneration'	)
							lowerCamelCase_  :    List[Any]   	=			'./tests/fixtures/tests_samples/COCO/000000039769.png'
							lowerCamelCase_  :    List[Any]   	=			'a photo of'

							lowerCamelCase_  :    Dict   	=			pipe(lowerCamelCase					,			prompt=lowerCamelCase	)
							self.assertTrue(outputs[0]['generated_text'].startswith(lowerCamelCase	)	)



					@slow
					@require_torch
					def      __a    (		self    :    Union[str, Any]	):
							lowerCamelCase_  :    List[str]   	=			pipeline('image-to-text'					,			model='ydshieh/vit-gpt2-coco-en'	)
							lowerCamelCase_  :    Any   	=			'./tests/fixtures/tests_samples/COCO/000000039769.png'

							lowerCamelCase_  :    str   	=			pipe(lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}]	)

							lowerCamelCase_  :    Union[str, Any]   	=			pipe([image, image]	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        [{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}],
							        [{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}],
							    ]					,			)



					@slow
					@require_torch
					def      __a    (		self    :    List[Any]	):
							lowerCamelCase_  :    int   	=			pipeline('image-to-text'					,			model='Salesforce/blip-image-captioning-base'	)
							lowerCamelCase_  :    Tuple   	=			'https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/pokemon.png'
							lowerCamelCase_  :    List[Any]   	=			Image.open(requests.get(lowerCamelCase					,			stream=lowerCamelCase	).raw	)

							lowerCamelCase_  :    Optional[Any]   	=			pipe(lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a pink pokemon pokemon with a blue shirt and a blue shirt'}]	)



					@slow
					@require_torch
					def      __a    (		self    :    List[str]	):
							lowerCamelCase_  :    Optional[Any]   	=			pipeline('image-to-text'					,			model='microsoft/git-base-coco'	)
							lowerCamelCase_  :    str   	=			'https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/pokemon.png'
							lowerCamelCase_  :    Dict   	=			Image.open(requests.get(lowerCamelCase					,			stream=lowerCamelCase	).raw	)

							lowerCamelCase_  :    Optional[int]   	=			pipe(lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a cartoon of a purple character.'}]	)



					@slow
					@require_torch
					def      __a    (		self    :    List[str]	):
							lowerCamelCase_  :    List[Any]   	=			pipeline('image-to-text'					,			model='Salesforce/blip-image-captioning-base'	)
							lowerCamelCase_  :    int   	=			'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg'
							lowerCamelCase_  :    List[str]   	=			Image.open(requests.get(lowerCamelCase					,			stream=lowerCamelCase	).raw	)

							lowerCamelCase_  :    int   	=			'a photography of'

							lowerCamelCase_  :    Tuple   	=			pipe(lowerCamelCase					,			prompt=lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a photography of a volcano'}]	)

							with self.assertRaises(lowerCamelCase	):
									lowerCamelCase_  :    Union[str, Any]   	=			pipe([image, image]					,			prompt=[prompt, prompt]	)



					@slow
					@require_torch
					def      __a    (		self    :    int	):
							lowerCamelCase_  :    Optional[int]   	=			pipeline('image-to-text'					,			model='microsoft/git-base-coco'	)
							lowerCamelCase_  :    int   	=			'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg'
							lowerCamelCase_  :    Optional[Any]   	=			Image.open(requests.get(lowerCamelCase					,			stream=lowerCamelCase	).raw	)

							lowerCamelCase_  :    Optional[int]   	=			'a photo of a'

							lowerCamelCase_  :    Optional[int]   	=			pipe(lowerCamelCase					,			prompt=lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a photo of a tent with a tent and a tent in the background.'}]	)

							with self.assertRaises(lowerCamelCase	):
									lowerCamelCase_  :    List[Any]   	=			pipe([image, image]					,			prompt=[prompt, prompt]	)



					@unittest.skipIf(
					    not is_torch_greater_or_equal_than_1_11					,			reason='`Pix2StructImageProcessor` requires `torch>=1.11.0`.'	)
					@slow
					@require_torch
					def      __a    (		self    :    Any	):
							lowerCamelCase_  :    Union[str, Any]   	=			pipeline('image-to-text'					,			model='google/pix2struct-ai2d-base'	)
							lowerCamelCase_  :    List[str]   	=			'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg'
							lowerCamelCase_  :    Tuple   	=			Image.open(requests.get(lowerCamelCase					,			stream=lowerCamelCase	).raw	)

							lowerCamelCase_  :    str   	=			'What does the label 15 represent? (1) lava (2) core (3) tunnel (4) ash cloud'

							lowerCamelCase_  :    Union[str, Any]   	=			pipe(lowerCamelCase					,			prompt=lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'ash cloud'}]	)

							with self.assertRaises(lowerCamelCase	):
									lowerCamelCase_  :    Optional[Any]   	=			pipe([image, image]					,			prompt=[prompt, prompt]	)





					@slow
					@require_tf
					def      __a    (		self    :    Dict	):
							lowerCamelCase_  :    List[str]   	=			pipeline('image-to-text'					,			model='ydshieh/vit-gpt2-coco-en'					,			framework='tf'	)
							lowerCamelCase_  :    Dict   	=			'./tests/fixtures/tests_samples/COCO/000000039769.png'

							lowerCamelCase_  :    Optional[Any]   	=			pipe(lowerCamelCase	)
							self.assertEqual(lowerCamelCase					,			[{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}]	)

							lowerCamelCase_  :    List[Any]   	=			pipe([image, image]	)
							self.assertEqual(
							    lowerCamelCase					,			[
							        [{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}],
							        [{'generated_text': 'a cat laying on a blanket next to a cat laying on a bed '}],
							    ]					,			)
