





"""simple docstring"""



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,
)


a_ =			{
    """configuration_albert""": ["""ALBERT_PRETRAINED_CONFIG_ARCHIVE_MAP""", """AlbertConfig""", """AlbertOnnxConfig"""],
}

try:
       if not is_sentencepiece_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a_ =			["""AlbertTokenizer"""]

try:
       if not is_tokenizers_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a_ =			["""AlbertTokenizerFast"""]

try:
       if not is_torch_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a_ =			[
           """ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST""",
           """AlbertForMaskedLM""",
           """AlbertForMultipleChoice""",
           """AlbertForPreTraining""",
           """AlbertForQuestionAnswering""",
           """AlbertForSequenceClassification""",
           """AlbertForTokenClassification""",
           """AlbertModel""",
           """AlbertPreTrainedModel""",
           """load_tf_weights_in_albert""",
       ]

try:
       if not is_tf_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a_ =			[
           """TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST""",
           """TFAlbertForMaskedLM""",
           """TFAlbertForMultipleChoice""",
           """TFAlbertForPreTraining""",
           """TFAlbertForQuestionAnswering""",
           """TFAlbertForSequenceClassification""",
           """TFAlbertForTokenClassification""",
           """TFAlbertMainLayer""",
           """TFAlbertModel""",
           """TFAlbertPreTrainedModel""",
       ]

try:
       if not is_flax_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a_ =			[
           """FlaxAlbertForMaskedLM""",
           """FlaxAlbertForMultipleChoice""",
           """FlaxAlbertForPreTraining""",
           """FlaxAlbertForQuestionAnswering""",
           """FlaxAlbertForSequenceClassification""",
           """FlaxAlbertForTokenClassification""",
           """FlaxAlbertModel""",
           """FlaxAlbertPreTrainedModel""",
       ]

if TYPE_CHECKING:
       from .configuration_albert import ALBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, AlbertConfig, AlbertOnnxConfig

       try:
              if not is_sentencepiece_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .tokenization_albert import AlbertTokenizer

       try:
              if not is_tokenizers_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .tokenization_albert_fast import AlbertTokenizerFast

       try:
              if not is_torch_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_albert import (
                  ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
                  AlbertForMaskedLM,
                  AlbertForMultipleChoice,
                  AlbertForPreTraining,
                  AlbertForQuestionAnswering,
                  AlbertForSequenceClassification,
                  AlbertForTokenClassification,
                  AlbertModel,
                  AlbertPreTrainedModel,
                  load_tf_weights_in_albert,
              )

       try:
              if not is_tf_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_tf_albert import (
                  TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
                  TFAlbertForMaskedLM,
                  TFAlbertForMultipleChoice,
                  TFAlbertForPreTraining,
                  TFAlbertForQuestionAnswering,
                  TFAlbertForSequenceClassification,
                  TFAlbertForTokenClassification,
                  TFAlbertMainLayer,
                  TFAlbertModel,
                  TFAlbertPreTrainedModel,
              )

       try:
              if not is_flax_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_flax_albert import (
                  FlaxAlbertForMaskedLM,
                  FlaxAlbertForMultipleChoice,
                  FlaxAlbertForPreTraining,
                  FlaxAlbertForQuestionAnswering,
                  FlaxAlbertForSequenceClassification,
                  FlaxAlbertForTokenClassification,
                  FlaxAlbertModel,
                  FlaxAlbertPreTrainedModel,
              )
else:
       import sys

       a_ =			_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)











"""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.
import torch

from ..models.speechta import SpeechTaForTextToSpeech, SpeechTaHifiGan, SpeechTaProcessor
from ..utils import is_datasets_available
from .base import PipelineTool


if is_datasets_available():
       from datasets import load_dataset






class 		__snake_case					(		SCREAMING_SNAKE_CASE__     ):





 """simple docstring"""


 _lowerCamelCase        =     """microsoft/speecht5_tts"""
 _lowerCamelCase        =     (
     """This is a tool that reads an English text out loud. It takes an input named `text` which should contain the """
     """text to read (in English) and returns a waveform object containing the sound."""
 )
 _lowerCamelCase        =     """text_reader"""
 _lowerCamelCase        =     SpeechTaProcessor
 _lowerCamelCase        =     SpeechTaForTextToSpeech
 _lowerCamelCase        =     SpeechTaHifiGan

 _lowerCamelCase        =     ["""text"""]
 _lowerCamelCase        =     ["""audio"""]

 def 					UpperCamelCase__(    self  ):



     '''simple docstring'''




     if self.post_processor is None:
         __A    :						List[str]						 =     '''microsoft/speecht5_hifigan'''
     super().setup()

 def 					UpperCamelCase__(    self					,						__lowerCamelCase					,						__lowerCamelCase=None  ):



     '''simple docstring'''




     __A    :						int						 =     self.pre_processor(text=__lowerCamelCase					,						return_tensors='''pt'''					,						truncation=__lowerCamelCase  )

     if speaker_embeddings is None:
         if not is_datasets_available():
             raise ImportError('''Datasets needs to be installed if not passing speaker embeddings.'''  )

         __A    :						List[Any]						 =     load_dataset('''Matthijs/cmu-arctic-xvectors'''					,						split='''validation'''  )
         __A    :						int						 =     torch.tensor(embeddings_dataset[7305]['''xvector''']  ).unsqueeze(0  )

     return {"input_ids": inputs["input_ids"], "speaker_embeddings": speaker_embeddings}

 def 					UpperCamelCase__(    self					,						__lowerCamelCase  ):



     '''simple docstring'''




     with torch.no_grad():
         return self.model.generate_speech(**__lowerCamelCase  )







 def 					UpperCamelCase__(    self					,						__lowerCamelCase  ):



     '''simple docstring'''




     with torch.no_grad():
         return self.post_processor(__lowerCamelCase  ).cpu().detach()




