



'''simple docstring'''
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__				(   snake_case_					):
 """simple docstring"""


 A__    :    Dict						=				{
     '''sld_warmup_steps''': 15,
     '''sld_guidance_scale''': 20,
     '''sld_threshold''': 0.0,
     '''sld_momentum_scale''': 0.0,
     '''sld_mom_beta''': 0.0,
 }
 A__    :    Dict						=				{
     '''sld_warmup_steps''': 10,
     '''sld_guidance_scale''': 1000,
     '''sld_threshold''': 0.01,
     '''sld_momentum_scale''': 0.3,
     '''sld_mom_beta''': 0.4,
 }
 A__    :    Optional[int]						=				{
     '''sld_warmup_steps''': 7,
     '''sld_guidance_scale''': 2000,
     '''sld_threshold''': 0.025,
     '''sld_momentum_scale''': 0.5,
     '''sld_mom_beta''': 0.7,
 }
 A__    :    Any						=				{
     '''sld_warmup_steps''': 0,
     '''sld_guidance_scale''': 5000,
     '''sld_threshold''': 1.0,
     '''sld_momentum_scale''': 0.5,
     '''sld_mom_beta''': 0.7,
 }







@dataclass
class   SCREAMING_SNAKE_CASE__				(   snake_case_					):
 """simple docstring"""


 A__    :    Union[List[PIL.Image.Image], np.ndarray]
 A__    :    Optional[List[bool]]
 A__    :    Optional[Union[List[PIL.Image.Image], np.ndarray]]
 A__    :    Optional[str]


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










'''simple docstring'''
from ..utils import DummyObject, requires_backends


class   SCREAMING_SNAKE_CASE__				(   metaclass=snake_case_					):
	"""simple docstring"""


	A__    :    Optional[int]						=				['''torch''', '''transformers''', '''invisible_watermark''']


	def __init__(		self					,   *A					,   **A			) ->							Tuple:
				requires_backends(self					,   ["""torch""", """transformers""", """invisible_watermark"""]			)


	@classmethod
	def      a__ (		cls					,   *A					,   **A			) ->							Optional[int]:
				requires_backends(cls					,   ["""torch""", """transformers""", """invisible_watermark"""]			)





	@classmethod
	def      a__ (		cls					,   *A					,   **A			) ->							List[Any]:
				requires_backends(cls					,   ["""torch""", """transformers""", """invisible_watermark"""]			)







class   SCREAMING_SNAKE_CASE__				(   metaclass=snake_case_					):
	"""simple docstring"""


	A__    :    Optional[int]						=				['''torch''', '''transformers''', '''invisible_watermark''']


	def __init__(		self					,   *A					,   **A			) ->							List[str]:
				requires_backends(self					,   ["""torch""", """transformers""", """invisible_watermark"""]			)


	@classmethod
	def      a__ (		cls					,   *A					,   **A			) ->							Union[str, Any]:
				requires_backends(cls					,   ["""torch""", """transformers""", """invisible_watermark"""]			)





	@classmethod
	def      a__ (		cls					,   *A					,   **A			) ->							Union[str, Any]:
				requires_backends(cls					,   ["""torch""", """transformers""", """invisible_watermark"""]			)





