


import unittest

from transformers import is_torch_available, load_tool

from .test_tools_common import ToolTesterMixin


if is_torch_available():
 import torch


class 		_SCREAMING_SNAKE_CASE    (	unittest.TestCase       ,  A    ):


     def 					__snake_case(     self						):
       _UpperCAmelCase		:       Dict       =    load_tool("""speech-to-text"""						)
       self.tool.setup()


     def 					__snake_case(     self						):
       _UpperCAmelCase		:       List[Any]       =    self.tool(torch.ones(30_00						)						)
       self.assertEqual(A_       ,				""" you"""						)




     def 					__snake_case(     self						):
       _UpperCAmelCase		:       int       =    self.tool(audio=torch.ones(30_00						)						)
       self.assertEqual(A_       ,				""" you"""						)









from dataclasses import dataclass
from enum import Enum
from typing import List, Optional, Union

import numpy as np
import PIL
from PIL import Image

from ...utils import BaseOutput, is_torch_available, is_transformers_available


@dataclass
class 		_SCREAMING_SNAKE_CASE    (	A    ):
     __SCREAMING_SNAKE_CASE    				=						{
         '''sld_warmup_steps''': 1_5,
         '''sld_guidance_scale''': 2_0,
         '''sld_threshold''': 0.0,
         '''sld_momentum_scale''': 0.0,
         '''sld_mom_beta''': 0.0,
     }
     __SCREAMING_SNAKE_CASE    				=						{
         '''sld_warmup_steps''': 1_0,
         '''sld_guidance_scale''': 1_0_0_0,
         '''sld_threshold''': 0.01,
         '''sld_momentum_scale''': 0.3,
         '''sld_mom_beta''': 0.4,
     }
     __SCREAMING_SNAKE_CASE    				=						{
         '''sld_warmup_steps''': 7,
         '''sld_guidance_scale''': 2_0_0_0,
         '''sld_threshold''': 0.025,
         '''sld_momentum_scale''': 0.5,
         '''sld_mom_beta''': 0.7,
     }
     __SCREAMING_SNAKE_CASE    				=						{
         '''sld_warmup_steps''': 0,
         '''sld_guidance_scale''': 5_0_0_0,
         '''sld_threshold''': 1.0,
         '''sld_momentum_scale''': 0.5,
         '''sld_mom_beta''': 0.7,
     }


@dataclass
class 		_SCREAMING_SNAKE_CASE    (	A    ):
     __SCREAMING_SNAKE_CASE    				=						42
     __SCREAMING_SNAKE_CASE    				=						42
     __SCREAMING_SNAKE_CASE    				=						42
     __SCREAMING_SNAKE_CASE    				=						42


if is_transformers_available() and is_torch_available():
 from .pipeline_stable_diffusion_safe import StableDiffusionPipelineSafe
 from .safety_checker import SafeStableDiffusionSafetyChecker





