


"""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 		_A    (    _a    ):







    """simple docstring"""




    UpperCAmelCase			:				List[str]      =			{
        """sld_warmup_steps""": 1_5,
        """sld_guidance_scale""": 2_0,
        """sld_threshold""": 0.0,
        """sld_momentum_scale""": 0.0,
        """sld_mom_beta""": 0.0,
    }
    UpperCAmelCase			:				Optional[Any]      =			{
        """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,
    }
    UpperCAmelCase			:				Tuple      =			{
        """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,
    }
    UpperCAmelCase			:				Any      =			{
        """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 		_A    (    _a    ):







    """simple docstring"""




    UpperCAmelCase			:				Union[List[PIL.Image.Image], np.ndarray]
    UpperCAmelCase			:				Optional[List[bool]]
    UpperCAmelCase			:				Optional[Union[List[PIL.Image.Image], np.ndarray]]
    UpperCAmelCase			:				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 		_A    (    metaclass=_a    ):







    """simple docstring"""




    UpperCAmelCase			:				int      =			["""torch""", """transformers""", """invisible_watermark"""]






    def __init__(   self							:       int							,					*__UpperCAmelCase							:       List[Any]							,					**__UpperCAmelCase							:       Optional[int]):
      requires_backends(self							,					["torch", "transformers", "invisible_watermark"])






    @classmethod
    def       __snake_case				(   cls							:       Optional[Any]							,					*__UpperCAmelCase							:       Any							,					**__UpperCAmelCase							:       int):
      requires_backends(cls							,					["torch", "transformers", "invisible_watermark"])






    @classmethod
    def       __snake_case				(   cls							:       List[Any]							,					*__UpperCAmelCase							:       Optional[Any]							,					**__UpperCAmelCase							:       Tuple):
      requires_backends(cls							,					["torch", "transformers", "invisible_watermark"])






class 		_A    (    metaclass=_a    ):







    """simple docstring"""




    UpperCAmelCase			:				Optional[int]      =			["""torch""", """transformers""", """invisible_watermark"""]






    def __init__(   self							:       int							,					*__UpperCAmelCase							:       Tuple							,					**__UpperCAmelCase							:       Tuple):
      requires_backends(self							,					["torch", "transformers", "invisible_watermark"])






    @classmethod
    def       __snake_case				(   cls							:       int							,					*__UpperCAmelCase							:       str							,					**__UpperCAmelCase							:       Optional[int]):
      requires_backends(cls							,					["torch", "transformers", "invisible_watermark"])






    @classmethod
    def       __snake_case				(   cls							:       Any							,					*__UpperCAmelCase							:       Optional[int]							,					**__UpperCAmelCase							:       Union[str, Any]):
      requires_backends(cls							,					["torch", "transformers", "invisible_watermark"])






