






"""simple docstring"""


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

import numpy as np
import PIL

from ...utils import BaseOutput, is_invisible_watermark_available, is_torch_available, is_transformers_available




@dataclass
class 	__UpperCamelCase    (    UpperCamelCase	):
					lowercase_						:							Union[List[PIL.Image.Image], np.ndarray]


if is_transformers_available() and is_torch_available() and is_invisible_watermark_available():
		from .pipeline_stable_diffusion_xl import StableDiffusionXLPipeline
		from .pipeline_stable_diffusion_xl_imgaimg import StableDiffusionXLImgaImgPipeline







"""simple docstring"""


import unittest

import torch

from diffusers import DDIMScheduler, TextToVideoZeroPipeline
from diffusers.utils import load_pt, require_torch_gpu, slow

from ..test_pipelines_common import assert_mean_pixel_difference




@slow
@require_torch_gpu
class 	__UpperCamelCase    (    unittest.TestCase	):
					def 					UpperCAmelCase__   (       self       :  Union[str, Any]      )  ->					Optional[Any]:
											lowerCAmelCase      :Union[str, Any]				  = 'runwayml/stable-diffusion-v1-5'
											lowerCAmelCase      :Tuple				  = TextToVideoZeroPipeline.from_pretrained(UpperCAmelCase    ,   torch_dtype=torch.floataa      ).to('cuda'      )
											lowerCAmelCase      :Union[str, Any]				  = DDIMScheduler.from_config(pipe.scheduler.config      )
											lowerCAmelCase      :str				  = torch.Generator(device='cuda'      ).manual_seed(0      )

											lowerCAmelCase      :Any				  = 'A bear is playing a guitar on Times Square'
											lowerCAmelCase      :Optional[Any]				  = pipe(prompt=UpperCAmelCase    ,   generator=UpperCAmelCase      ).images

											lowerCAmelCase      :List[str]				  = load_pt(
											    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/text-to-video/A bear is playing a guitar on Times Square.pt'      )

											assert_mean_pixel_difference(UpperCAmelCase    ,   UpperCAmelCase      )