def lowerCamelCase  (   ):
    '''simple docstring'''







    for n in range(1  ,			1_000_000	):
        yield n * (n + 1) // 2



def lowerCamelCase  (   SCREAMING_SNAKE_CASE	):
    '''simple docstring'''







    __UpperCamelCase     :Tuple			  =		1
    __UpperCamelCase     :int			  =		2
    while i * i <= n:
        __UpperCamelCase     :Union[str, Any]			  =		0
        while n % i == 0:
            n //= i
            multiplicity += 1
        divisors_count *= multiplicity + 1
        i += 1
    if n > 1:
        divisors_count *= 2
    return divisors_count



def lowerCamelCase  (   ):
    '''simple docstring'''







    return next(i for i in triangle_number_generator() if count_divisors(SCREAMING_SNAKE_CASE__	) > 500	)


if __name__ == "__main__":
    print(solution())







# 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		(      lowerCAmelCase     ):
    _a		:						Union[str, Any]=				"microsoft/speecht5_tts"
    _a		:						Tuple=				(
        "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."
    )
    _a		:						Dict=				"text_reader"
    _a		:						Optional[Any]=				SpeechTaProcessor
    _a		:						Tuple=				SpeechTaForTextToSpeech
    _a		:						Optional[int]=				SpeechTaHifiGan

    _a		:						Union[str, Any]=				["text"]
    _a		:						Optional[int]=				["audio"]
    def       _SCREAMING_SNAKE_CASE       (  self      ):



           '''simple docstring'''







           if self.post_processor is None:
                  lowercase							:     Any       	=      """microsoft/speecht5_hifigan"""
           super().setup()
    def       _SCREAMING_SNAKE_CASE       (  self		,snake_case		,snake_case=None      ):



           '''simple docstring'''







           lowercase							:     int       	=      self.pre_processor(text=snake_case		,return_tensors="""pt"""		,truncation=snake_case      )

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

                  lowercase							:     Tuple       	=      load_dataset("""Matthijs/cmu-arctic-xvectors"""		,split="""validation"""      )
                  lowercase							:     List[str]       	=      torch.tensor(embeddings_dataset[7305]["""xvector"""]      ).unsqueeze(0      )

           return {"input_ids": inputs["input_ids"], "speaker_embeddings": speaker_embeddings}
    def       _SCREAMING_SNAKE_CASE       (  self		,snake_case      ):



           '''simple docstring'''







           with torch.no_grad():
                  return self.model.generate_speech(**snake_case      )
    def       _SCREAMING_SNAKE_CASE       (  self		,snake_case      ):



           '''simple docstring'''







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