


from ...configuration_utils import PretrainedConfig
from ...utils import logging


SCREAMING_SNAKE_CASE							    =							logging.get_logger(__name__)

SCREAMING_SNAKE_CASE							    =							{
    'facebook/dpr-ctx_encoder-single-nq-base': (
        'https://huggingface.co/facebook/dpr-ctx_encoder-single-nq-base/resolve/main/config.json'
    ),
    'facebook/dpr-question_encoder-single-nq-base': (
        'https://huggingface.co/facebook/dpr-question_encoder-single-nq-base/resolve/main/config.json'
    ),
    'facebook/dpr-reader-single-nq-base': (
        'https://huggingface.co/facebook/dpr-reader-single-nq-base/resolve/main/config.json'
    ),
    'facebook/dpr-ctx_encoder-multiset-base': (
        'https://huggingface.co/facebook/dpr-ctx_encoder-multiset-base/resolve/main/config.json'
    ),
    'facebook/dpr-question_encoder-multiset-base': (
        'https://huggingface.co/facebook/dpr-question_encoder-multiset-base/resolve/main/config.json'
    ),
    'facebook/dpr-reader-multiset-base': (
        'https://huggingface.co/facebook/dpr-reader-multiset-base/resolve/main/config.json'
    ),
}


class     __UpperCAmelCase							(       __A							):


     """simple docstring"""
     _lowerCamelCase   =      """dpr"""


     def __init__(	self  ,       __A=30522  ,       __A=768  ,       __A=12  ,       __A=12  ,       __A=3072  ,       __A="gelu"  ,       __A=0.1  ,       __A=0.1  ,       __A=512  ,       __A=2  ,       __A=0.02  ,       __A=1E-12  ,       __A=0  ,       __A="absolute"  ,       __A = 0  ,       **__A  ,       ):
        super().__init__(pad_token_id=__A  ,       **__A    )

        __a					=  vocab_size
        __a					=  hidden_size
        __a					=  num_hidden_layers
        __a					=  num_attention_heads
        __a					=  hidden_act
        __a					=  intermediate_size
        __a					=  hidden_dropout_prob
        __a					=  attention_probs_dropout_prob
        __a					=  max_position_embeddings
        __a					=  type_vocab_size
        __a					=  initializer_range
        __a					=  layer_norm_eps
        __a					=  projection_dim
        __a					=  position_embedding_type











'''simple docstring'''







# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from ..models.whisper import WhisperForConditionalGeneration, WhisperProcessor
from .base import PipelineTool





class 						SCREAMING_SNAKE_CASE       (       SCREAMING_SNAKE_CASE_							):
			snake_case__	=					"openai/whisper-base"
			snake_case__	=					(
			    "This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the "
			    "transcribed text."
			)
			snake_case__	=					"transcriber"
			snake_case__	=					WhisperProcessor
			snake_case__	=					WhisperForConditionalGeneration

			snake_case__	=					["audio"]
			snake_case__	=					["text"]






			def       SCREAMING_SNAKE_CASE   (			self   :      str	,    __SCREAMING_SNAKE_CASE   :      Tuple       )      ->      str:
							return self.pre_processor(__SCREAMING_SNAKE_CASE	,    return_tensors='''pt'''       ).input_features






			def       SCREAMING_SNAKE_CASE   (			self   :      List[Any]	,    __SCREAMING_SNAKE_CASE   :      Optional[Any]       )      ->      List[str]:
							return self.model.generate(inputs=__SCREAMING_SNAKE_CASE       )







			def       SCREAMING_SNAKE_CASE   (			self   :      Dict	,    __SCREAMING_SNAKE_CASE   :      Optional[int]       )      ->      Optional[Any]:
							return self.pre_processor.batch_decode(__SCREAMING_SNAKE_CASE	,    skip_special_tokens=__SCREAMING_SNAKE_CASE       )[0]




