






'''simple docstring'''

import unittest

from transformers import BlenderbotTokenizer, BlenderbotTokenizerFast
from transformers.utils import cached_property
class        __lowerCAmelCase  (					unittest.TestCase				):






       """simple docstring"""





       @cached_property
       def snake_case__		(							self     :	int  )					->  Dict:

            '''simple docstring'''




            return BlenderbotTokenizer.from_pretrained('''facebook/blenderbot-3B'''  )





       @cached_property
       def snake_case__		(							self     :	Optional[Any]  )					->  Union[str, Any]:

            '''simple docstring'''




            return BlenderbotTokenizerFast.from_pretrained('''facebook/blenderbot-3B'''  )





       def snake_case__		(							self     :	List[str]  )					->  Optional[int]:

            '''simple docstring'''




            _UpperCamelCase       					= self.tokenizer_ab
            _UpperCamelCase       					= ''' I am a small frog.'''
            _UpperCamelCase       					= tok([src_text] ,				padding=lowerCAmelCase__ ,				truncation=lowerCAmelCase__  )['''input_ids''']
            _UpperCamelCase       					= tok.batch_decode(lowerCAmelCase__ ,				skip_special_tokens=lowerCAmelCase__ ,				clean_up_tokenization_spaces=lowerCAmelCase__  )[0]
            assert src_text == decoded





       def snake_case__		(							self     :	int  )					->  Optional[Any]:

            '''simple docstring'''




            _UpperCamelCase       					= self.rust_tokenizer_ab
            _UpperCamelCase       					= ''' I am a small frog.'''
            _UpperCamelCase       					= tok([src_text] ,				padding=lowerCAmelCase__ ,				truncation=lowerCAmelCase__  )['''input_ids''']
            _UpperCamelCase       					= tok.batch_decode(lowerCAmelCase__ ,				skip_special_tokens=lowerCAmelCase__ ,				clean_up_tokenization_spaces=lowerCAmelCase__  )[0]
            assert src_text == decoded





       def snake_case__		(							self     :	Tuple  )					->  Any:

            '''simple docstring'''




            assert self.tokenizer_ab.add_prefix_space
            assert self.tokenizer_ab([''' Sam''', '''Sam''']  ).input_ids == [[5502, 2], [5502, 2]]







       def snake_case__		(							self     :	Dict  )					->  List[str]:

            '''simple docstring'''




            assert self.rust_tokenizer_ab.add_prefix_space
            assert self.rust_tokenizer_ab([''' Sam''', '''Sam''']  ).input_ids == [[5502, 2], [5502, 2]]










'''simple docstring'''

from ..utils import (
    OptionalDependencyNotAvailable,
    is_flax_available,
    is_invisible_watermark_available,
    is_k_diffusion_available,
    is_librosa_available,
    is_note_seq_available,
    is_onnx_available,
    is_torch_available,
    is_transformers_available,
)


try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_pt_objects import *  # noqa F403
else:
   from .consistency_models import ConsistencyModelPipeline
   from .dance_diffusion import DanceDiffusionPipeline
   from .ddim import DDIMPipeline
   from .ddpm import DDPMPipeline
   from .dit import DiTPipeline
   from .latent_diffusion import LDMSuperResolutionPipeline
   from .latent_diffusion_uncond import LDMPipeline
   from .pipeline_utils import AudioPipelineOutput, DiffusionPipeline, ImagePipelineOutput
   from .pndm import PNDMPipeline
   from .repaint import RePaintPipeline
   from .score_sde_ve import ScoreSdeVePipeline
   from .stochastic_karras_ve import KarrasVePipeline

try:
   if not (is_torch_available() and is_librosa_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_torch_and_librosa_objects import *  # noqa F403
else:
   from .audio_diffusion import AudioDiffusionPipeline, Mel

try:
   if not (is_torch_available() and is_transformers_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
   from .alt_diffusion import AltDiffusionImgaImgPipeline, AltDiffusionPipeline
   from .audioldm import AudioLDMPipeline
   from .controlnet import (
       StableDiffusionControlNetImgaImgPipeline,
       StableDiffusionControlNetInpaintPipeline,
       StableDiffusionControlNetPipeline,
   )
   from .deepfloyd_if import (
       IFImgaImgPipeline,
       IFImgaImgSuperResolutionPipeline,
       IFInpaintingPipeline,
       IFInpaintingSuperResolutionPipeline,
       IFPipeline,
       IFSuperResolutionPipeline,
   )
   from .kandinsky import (
       KandinskyImgaImgPipeline,
       KandinskyInpaintPipeline,
       KandinskyPipeline,
       KandinskyPriorPipeline,
   )
   from .kandinskya_a import (
       KandinskyVaaControlnetImgaImgPipeline,
       KandinskyVaaControlnetPipeline,
       KandinskyVaaImgaImgPipeline,
       KandinskyVaaInpaintPipeline,
       KandinskyVaaPipeline,
       KandinskyVaaPriorEmbaEmbPipeline,
       KandinskyVaaPriorPipeline,
   )
   from .latent_diffusion import LDMTextToImagePipeline
   from .paint_by_example import PaintByExamplePipeline
   from .semantic_stable_diffusion import SemanticStableDiffusionPipeline
   from .shap_e import ShapEImgaImgPipeline, ShapEPipeline
   from .stable_diffusion import (
       CycleDiffusionPipeline,
       StableDiffusionAttendAndExcitePipeline,
       StableDiffusionDepthaImgPipeline,
       StableDiffusionDiffEditPipeline,
       StableDiffusionImageVariationPipeline,
       StableDiffusionImgaImgPipeline,
       StableDiffusionInpaintPipeline,
       StableDiffusionInpaintPipelineLegacy,
       StableDiffusionInstructPixaPixPipeline,
       StableDiffusionLatentUpscalePipeline,
       StableDiffusionLDMaDPipeline,
       StableDiffusionModelEditingPipeline,
       StableDiffusionPanoramaPipeline,
       StableDiffusionParadigmsPipeline,
       StableDiffusionPipeline,
       StableDiffusionPixaPixZeroPipeline,
       StableDiffusionSAGPipeline,
       StableDiffusionUpscalePipeline,
       StableUnCLIPImgaImgPipeline,
       StableUnCLIPPipeline,
   )
   from .stable_diffusion_safe import StableDiffusionPipelineSafe
   from .text_to_video_synthesis import TextToVideoSDPipeline, TextToVideoZeroPipeline, VideoToVideoSDPipeline
   from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
   from .unidiffuser import ImageTextPipelineOutput, UniDiffuserModel, UniDiffuserPipeline, UniDiffuserTextDecoder
   from .versatile_diffusion import (
       VersatileDiffusionDualGuidedPipeline,
       VersatileDiffusionImageVariationPipeline,
       VersatileDiffusionPipeline,
       VersatileDiffusionTextToImagePipeline,
   )
   from .vq_diffusion import VQDiffusionPipeline


try:
   if not (is_torch_available() and is_transformers_available() and is_invisible_watermark_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_torch_and_transformers_and_invisible_watermark_objects import *  # noqa F403
else:
   from .stable_diffusion_xl import StableDiffusionXLImgaImgPipeline, StableDiffusionXLPipeline

try:
   if not is_onnx_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_onnx_objects import *  # noqa F403
else:
   from .onnx_utils import OnnxRuntimeModel

try:
   if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_torch_and_transformers_and_onnx_objects import *  # noqa F403
else:
   from .stable_diffusion import (
       OnnxStableDiffusionImgaImgPipeline,
       OnnxStableDiffusionInpaintPipeline,
       OnnxStableDiffusionInpaintPipelineLegacy,
       OnnxStableDiffusionPipeline,
       OnnxStableDiffusionUpscalePipeline,
       StableDiffusionOnnxPipeline,
   )

try:
   if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
else:
   from .stable_diffusion import StableDiffusionKDiffusionPipeline

try:
   if not is_flax_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_flax_objects import *  # noqa F403
else:
   from .pipeline_flax_utils import FlaxDiffusionPipeline


try:
   if not (is_flax_available() and is_transformers_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_flax_and_transformers_objects import *  # noqa F403
else:
   from .controlnet import FlaxStableDiffusionControlNetPipeline
   from .stable_diffusion import (
       FlaxStableDiffusionImgaImgPipeline,
       FlaxStableDiffusionInpaintPipeline,
       FlaxStableDiffusionPipeline,
   )
try:
   if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   from ..utils.dummy_transformers_and_torch_and_note_seq_objects import *  # noqa F403
else:
   from .spectrogram_diffusion import MidiProcessor, SpectrogramDiffusionPipeline



