

"""simple docstring"""





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


_a       =							logging.get_logger(__name__)

_a       =							{
    'RWKV/rwkv-4-169m-pile': 'https://huggingface.co/RWKV/rwkv-4-169m-pile/resolve/main/config.json',
    'RWKV/rwkv-4-430m-pile': 'https://huggingface.co/RWKV/rwkv-4-430m-pile/resolve/main/config.json',
    'RWKV/rwkv-4-1b5-pile': 'https://huggingface.co/RWKV/rwkv-4-1b5-pile/resolve/main/config.json',
    'RWKV/rwkv-4-3b-pile': 'https://huggingface.co/RWKV/rwkv-4-3b-pile/resolve/main/config.json',
    'RWKV/rwkv-4-7b-pile': 'https://huggingface.co/RWKV/rwkv-4-7b-pile/resolve/main/config.json',
    'RWKV/rwkv-4-14b-pile': 'https://huggingface.co/RWKV/rwkv-4-14b-pile/resolve/main/config.json',
    'RWKV/rwkv-raven-1b5': 'https://huggingface.co/RWKV/rwkv-raven-1b5/resolve/main/config.json',
    'RWKV/rwkv-raven-3b': 'https://huggingface.co/RWKV/rwkv-raven-3b/resolve/main/config.json',
    'RWKV/rwkv-raven-7b': 'https://huggingface.co/RWKV/rwkv-raven-7b/resolve/main/config.json',
    'RWKV/rwkv-raven-14b': 'https://huggingface.co/RWKV/rwkv-raven-14b/resolve/main/config.json',
}




class 					_lowerCAmelCase  (	lowercase   ):







 """simple docstring"""







 __UpperCAmelCase	:					List[str]   		=	"rwkv"
 __UpperCAmelCase	:					List[Any]   		=	{"max_position_embeddings": "context_length"}


 def __init__(			self							:			Optional[Any],					UpperCAmelCase__							:			Any=5_0_2_7_7,					UpperCAmelCase__							:			Union[str, Any]=1_0_2_4,					UpperCAmelCase__							:			int=4_0_9_6,					UpperCAmelCase__							:			str=3_2,					UpperCAmelCase__							:			int=None,					UpperCAmelCase__							:			Optional[int]=None,					UpperCAmelCase__							:			Optional[int]=1E-5,					UpperCAmelCase__							:			str=0,					UpperCAmelCase__							:			Optional[int]=0,					UpperCAmelCase__							:			Optional[int]=6,					UpperCAmelCase__							:			Any=False,					UpperCAmelCase__							:			Any=True,					**UpperCAmelCase__							:			int,					):
    __lowercase								=						vocab_size
    __lowercase								=						context_length
    __lowercase								=						hidden_size
    __lowercase								=						num_hidden_layers
    __lowercase								=						attention_hidden_size if attention_hidden_size is not None else hidden_size
    __lowercase								=						intermediate_size if intermediate_size is not None else 4 * hidden_size
    __lowercase								=						layer_norm_epsilon
    __lowercase								=						rescale_every
    __lowercase								=						use_cache

    __lowercase								=						bos_token_id
    __lowercase								=						eos_token_id

    super().__init__(
        tie_word_embeddings=UpperCAmelCase__,					bos_token_id=UpperCAmelCase__,					eos_token_id=UpperCAmelCase__,					**UpperCAmelCase__	)







"""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 					_lowerCAmelCase  (	lowercase   ):







 """simple docstring"""







 __UpperCAmelCase	:					Tuple   		=	"openai/whisper-base"
 __UpperCAmelCase	:					Union[str, Any]   		=	(
     "This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the "
     "transcribed text."
 )
 __UpperCAmelCase	:					List[str]   		=	"transcriber"
 __UpperCAmelCase	:					Optional[Any]   		=	WhisperProcessor
 __UpperCAmelCase	:					str   		=	WhisperForConditionalGeneration

 __UpperCAmelCase	:					List[str]   		=	["audio"]
 __UpperCAmelCase	:					Tuple   		=	["text"]


 def 					_lowercase (			self							:			str,					UpperCAmelCase__							:			int	):
    return self.pre_processor(UpperCAmelCase__,					return_tensors="pt"	).input_features


 def 					_lowercase (			self							:			Union[str, Any],					UpperCAmelCase__							:			Optional[Any]	):
    return self.model.generate(inputs=UpperCAmelCase__	)



 def 					_lowercase (			self							:			Dict,					UpperCAmelCase__							:			Optional[int]	):
    return self.pre_processor.batch_decode(UpperCAmelCase__,					skip_special_tokens=UpperCAmelCase__	)[0]




