



"""simple docstring"""







from typing import TYPE_CHECKING

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


lowerCAmelCase_   						=   {
    'configuration_owlvit': [
        'OWLVIT_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'OwlViTConfig',
        'OwlViTOnnxConfig',
        'OwlViTTextConfig',
        'OwlViTVisionConfig',
    ],
    'processing_owlvit': ['OwlViTProcessor'],
}


try:
							if not is_vision_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							pass
else:
							lowerCAmelCase_   						=   ['OwlViTFeatureExtractor']
							lowerCAmelCase_   						=   ['OwlViTImageProcessor']

try:
							if not is_torch_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							pass
else:
							lowerCAmelCase_   						=   [
							    'OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST',
							    'OwlViTModel',
							    'OwlViTPreTrainedModel',
							    'OwlViTTextModel',
							    'OwlViTVisionModel',
							    'OwlViTForObjectDetection',
							]

if TYPE_CHECKING:
							from .configuration_owlvit import (
							    OWLVIT_PRETRAINED_CONFIG_ARCHIVE_MAP,
							    OwlViTConfig,
							    OwlViTOnnxConfig,
							    OwlViTTextConfig,
							    OwlViTVisionConfig,
							)
							from .processing_owlvit import OwlViTProcessor

							try:
														if not is_vision_available():
																					raise OptionalDependencyNotAvailable()
							except OptionalDependencyNotAvailable:
														pass
							else:
														from .feature_extraction_owlvit import OwlViTFeatureExtractor
														from .image_processing_owlvit import OwlViTImageProcessor

							try:
														if not is_torch_available():
																					raise OptionalDependencyNotAvailable()
							except OptionalDependencyNotAvailable:
														pass
							else:
														from .modeling_owlvit import (
														    OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
														    OwlViTForObjectDetection,
														    OwlViTModel,
														    OwlViTPreTrainedModel,
														    OwlViTTextModel,
														    OwlViTVisionModel,
														)

else:
							import sys

							lowerCAmelCase_   						=   _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)


from typing import Dict

from .base import GenericTensor, Pipeline







class 			__A( a			):




				def 				SCREAMING_SNAKE_CASE_		(     self		,     _snake_case=None		,     _snake_case=None		,     _snake_case=None		,     **_snake_case				)    ->      Optional[Any]:


								'''simple docstring'''




								if tokenize_kwargs is None:
												__a										=    {}

								if truncation is not None:
												if "truncation" in tokenize_kwargs:
																raise ValueError(
																    '''truncation parameter defined twice (given as keyword argument as well as in tokenize_kwargs)'''				)
												__a										=    truncation

								__a										=    tokenize_kwargs

								__a										=    {}
								if return_tensors is not None:
												__a										=    return_tensors

								return preprocess_params, {}, postprocess_params




				def 				SCREAMING_SNAKE_CASE_		(     self		,     _snake_case		,     **_snake_case				)    ->      Dict[str, GenericTensor]:


								'''simple docstring'''




								__a										=    self.framework
								__a										=    self.tokenizer(_snake_case		,     return_tensors=_snake_case		,     **_snake_case				)
								return model_inputs




				def 				SCREAMING_SNAKE_CASE_		(     self		,     _snake_case				)    ->      Optional[Any]:


								'''simple docstring'''




								__a										=    self.model(**_snake_case				)
								return model_outputs




				def 				SCREAMING_SNAKE_CASE_		(     self		,     _snake_case		,     _snake_case=False				)    ->      Optional[int]:


								'''simple docstring'''




								if return_tensors:
												return model_outputs[0]
								if self.framework == "pt":
												return model_outputs[0].tolist()
								elif self.framework == "tf":
												return model_outputs[0].numpy().tolist()




				def __call__(     self		,     *_snake_case		,     **_snake_case				)    ->      Any:


								'''simple docstring'''




								return super().__call__(*_snake_case		,     **_snake_case				)