





"""simple docstring"""





from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_sentencepiece_available,
    is_speech_available,
    is_tf_available,
    is_torch_available,
)


__A											=						{
    'configuration_speech_to_text': ['SPEECH_TO_TEXT_PRETRAINED_CONFIG_ARCHIVE_MAP', 'Speech2TextConfig'],
    'processing_speech_to_text': ['Speech2TextProcessor'],
}

try:
  if not is_sentencepiece_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __A											=						['Speech2TextTokenizer']

try:
  if not is_speech_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __A											=						['Speech2TextFeatureExtractor']

try:
  if not is_tf_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __A											=						[
      'TF_SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST',
      'TFSpeech2TextForConditionalGeneration',
      'TFSpeech2TextModel',
      'TFSpeech2TextPreTrainedModel',
  ]

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __A											=						[
      'SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST',
      'Speech2TextForConditionalGeneration',
      'Speech2TextModel',
      'Speech2TextPreTrainedModel',
  ]


if TYPE_CHECKING:
  from .configuration_speech_to_text import SPEECH_TO_TEXT_PRETRAINED_CONFIG_ARCHIVE_MAP, SpeechaTextConfig
  from .processing_speech_to_text import SpeechaTextProcessor

  try:
    if not is_sentencepiece_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_speech_to_text import SpeechaTextTokenizer

  try:
    if not is_speech_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .feature_extraction_speech_to_text import SpeechaTextFeatureExtractor

  try:
    if not is_tf_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_tf_speech_to_text import (
        TF_SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST,
        TFSpeechaTextForConditionalGeneration,
        TFSpeechaTextModel,
        TFSpeechaTextPreTrainedModel,
    )

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_speech_to_text import (
        SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST,
        SpeechaTextForConditionalGeneration,
        SpeechaTextModel,
        SpeechaTextPreTrainedModel,
    )

else:
  import sys

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






'''simple docstring'''


import gc
import unittest

import numpy as np
import torch

from diffusers import StableDiffusionKDiffusionPipeline
from diffusers.utils import slow, torch_device
from diffusers.utils.testing_utils import enable_full_determinism, require_torch_gpu


enable_full_determinism()

@slow
@require_torch_gpu
class      A       (						unittest.TestCase       ):







							def   __lowerCAmelCase     (    self			:    int   )     ->				Any:


														"""simple docstring"""


														super().tearDown()
														gc.collect()
														torch.cuda.empty_cache()







							def   __lowerCAmelCase     (    self			:    List[Any]   )     ->				int:


														"""simple docstring"""


														_a      	=							StableDiffusionKDiffusionPipeline.from_pretrained('''CompVis/stable-diffusion-v1-4'''   )
														_a      	=							sd_pipe.to(lowerCAmelCase_   )
														sd_pipe.set_progress_bar_config(disable=lowerCAmelCase_   )

														sd_pipe.set_scheduler('''sample_euler'''   )

														_a      	=							'''A painting of a squirrel eating a burger'''
														_a      	=							torch.manual_seed(0   )
														_a      	=							sd_pipe([prompt]				,     generator=lowerCAmelCase_				,     guidance_scale=9.0				,     num_inference_steps=20				,     output_type='''np'''   )

														_a      	=							output.images

														_a      	=							image[0, -3:, -3:, -1]

														assert image.shape == (1, 5_12, 5_12, 3)
														_a      	=							np.array([0.0_4_4_7, 0.0_4_9_2, 0.0_4_6_8, 0.0_4_0_8, 0.0_3_8_3, 0.0_4_0_8, 0.0_3_5_4, 0.0_3_8_0, 0.0_3_3_9]   )

														assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-2







							def   __lowerCAmelCase     (    self			:    Any   )     ->				Optional[Any]:


														"""simple docstring"""


														_a      	=							StableDiffusionKDiffusionPipeline.from_pretrained('''stabilityai/stable-diffusion-2-1-base'''   )
														_a      	=							sd_pipe.to(lowerCAmelCase_   )
														sd_pipe.set_progress_bar_config(disable=lowerCAmelCase_   )

														sd_pipe.set_scheduler('''sample_euler'''   )

														_a      	=							'''A painting of a squirrel eating a burger'''
														_a      	=							torch.manual_seed(0   )
														_a      	=							sd_pipe([prompt]				,     generator=lowerCAmelCase_				,     guidance_scale=9.0				,     num_inference_steps=20				,     output_type='''np'''   )

														_a      	=							output.images

														_a      	=							image[0, -3:, -3:, -1]

														assert image.shape == (1, 5_12, 5_12, 3)
														_a      	=							np.array([0.1_2_3_7, 0.1_3_2_0, 0.1_4_3_8, 0.1_3_5_9, 0.1_3_9_0, 0.1_1_3_2, 0.1_2_7_7, 0.1_1_7_5, 0.1_1_1_2]   )

														assert np.abs(image_slice.flatten() - expected_slice   ).max() < 5e-1







							def   __lowerCAmelCase     (    self			:    Dict   )     ->				Optional[Any]:


														"""simple docstring"""


														_a      	=							StableDiffusionKDiffusionPipeline.from_pretrained('''stabilityai/stable-diffusion-2-1-base'''   )
														_a      	=							sd_pipe.to(lowerCAmelCase_   )
														sd_pipe.set_progress_bar_config(disable=lowerCAmelCase_   )

														sd_pipe.set_scheduler('''sample_dpmpp_2m'''   )

														_a      	=							'''A painting of a squirrel eating a burger'''
														_a      	=							torch.manual_seed(0   )
														_a      	=							sd_pipe(
														    [prompt]				,     generator=lowerCAmelCase_				,     guidance_scale=7.5				,     num_inference_steps=15				,     output_type='''np'''				,     use_karras_sigmas=lowerCAmelCase_				,     )

														_a      	=							output.images

														_a      	=							image[0, -3:, -3:, -1]

														assert image.shape == (1, 5_12, 5_12, 3)
														_a      	=							np.array(
														    [0.1_1_3_8_1_6_8_9, 0.1_2_1_1_2_9_2_1, 0.1_3_8_9_4_5_7, 0.1_2_5_4_9_6_0_6, 0.1_2_4_4_9_6_4, 0.1_0_8_3_1_5_1_7, 0.1_1_5_6_2_8_6_6, 0.1_0_8_6_7_8_1_6, 0.1_0_4_9_9_0_4_8]   )

														assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-2

