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

import numpy as np
import PIL
from PIL import Image

from ...utils import (
    BaseOutput,
    OptionalDependencyNotAvailable,
    is_flax_available,
    is_k_diffusion_available,
    is_k_diffusion_version,
    is_onnx_available,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)




@dataclass
class    SCREAMING_SNAKE_CASE__     (	UpperCAmelCase					):





	'''simple docstring'''



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


try:
			if not (is_transformers_available() and is_torch_available()):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ...utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
			from .pipeline_cycle_diffusion import CycleDiffusionPipeline
			from .pipeline_stable_diffusion import StableDiffusionPipeline
			from .pipeline_stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline
			from .pipeline_stable_diffusion_imgaimg import StableDiffusionImgaImgPipeline
			from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
			from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy
			from .pipeline_stable_diffusion_instruct_pixapix import StableDiffusionInstructPixaPixPipeline
			from .pipeline_stable_diffusion_latent_upscale import StableDiffusionLatentUpscalePipeline
			from .pipeline_stable_diffusion_ldmad import StableDiffusionLDMaDPipeline
			from .pipeline_stable_diffusion_model_editing import StableDiffusionModelEditingPipeline
			from .pipeline_stable_diffusion_panorama import StableDiffusionPanoramaPipeline
			from .pipeline_stable_diffusion_paradigms import StableDiffusionParadigmsPipeline
			from .pipeline_stable_diffusion_sag import StableDiffusionSAGPipeline
			from .pipeline_stable_diffusion_upscale import StableDiffusionUpscalePipeline
			from .pipeline_stable_unclip import StableUnCLIPPipeline
			from .pipeline_stable_unclip_imgaimg import StableUnCLIPImgaImgPipeline
			from .safety_checker import StableDiffusionSafetyChecker
			from .stable_unclip_image_normalizer import StableUnCLIPImageNormalizer

try:
			if not (is_transformers_available() and is_torch_available() and is_transformers_version('''>=''', '''4.25.0''')):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ...utils.dummy_torch_and_transformers_objects import StableDiffusionImageVariationPipeline
else:
			from .pipeline_stable_diffusion_image_variation import StableDiffusionImageVariationPipeline


try:
			if not (is_transformers_available() and is_torch_available() and is_transformers_version('''>=''', '''4.26.0''')):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ...utils.dummy_torch_and_transformers_objects import (
			    StableDiffusionDepthaImgPipeline,
			    StableDiffusionDiffEditPipeline,
			    StableDiffusionPixaPixZeroPipeline,
			)
else:
			from .pipeline_stable_diffusion_depthaimg import StableDiffusionDepthaImgPipeline
			from .pipeline_stable_diffusion_diffedit import StableDiffusionDiffEditPipeline
			from .pipeline_stable_diffusion_pixapix_zero import StableDiffusionPixaPixZeroPipeline


try:
			if not (
			    is_torch_available()
			    and is_transformers_available()
			    and is_k_diffusion_available()
			    and is_k_diffusion_version('''>=''', '''0.0.12''')
			):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ...utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
else:
			from .pipeline_stable_diffusion_k_diffusion import StableDiffusionKDiffusionPipeline

try:
			if not (is_transformers_available() and is_onnx_available()):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ...utils.dummy_onnx_objects import *  # noqa F403
else:
			from .pipeline_onnx_stable_diffusion import OnnxStableDiffusionPipeline, StableDiffusionOnnxPipeline
			from .pipeline_onnx_stable_diffusion_imgaimg import OnnxStableDiffusionImgaImgPipeline
			from .pipeline_onnx_stable_diffusion_inpaint import OnnxStableDiffusionInpaintPipeline
			from .pipeline_onnx_stable_diffusion_inpaint_legacy import OnnxStableDiffusionInpaintPipelineLegacy
			from .pipeline_onnx_stable_diffusion_upscale import OnnxStableDiffusionUpscalePipeline

if is_transformers_available() and is_flax_available():
			import flax




			@flax.struct.dataclass
			class    SCREAMING_SNAKE_CASE__     (	UpperCAmelCase					):





				'''simple docstring'''



				_UpperCAmelCase       :		np.ndarray
				_UpperCAmelCase       :		List[bool]




			from ...schedulers.scheduling_pndm_flax import PNDMSchedulerState
			from .pipeline_flax_stable_diffusion import FlaxStableDiffusionPipeline
			from .pipeline_flax_stable_diffusion_imgaimg import FlaxStableDiffusionImgaImgPipeline
			from .pipeline_flax_stable_diffusion_inpaint import FlaxStableDiffusionInpaintPipeline
			from .safety_checker_flax import FlaxStableDiffusionSafetyChecker

import os
import tempfile
import unittest
from pathlib import Path

from transformers import AutoConfig, is_torch_available
from transformers.testing_utils import require_torch, torch_device


if is_torch_available():
			from transformers import PyTorchBenchmark, PyTorchBenchmarkArguments




@require_torch
class    SCREAMING_SNAKE_CASE__     (	unittest.TestCase					):





	'''simple docstring'''



	def  A							(  self	:    List[Any]			,  lowercase	:    Dict	):


				'''simple docstring'''


				for model_result in results.values():
							for batch_size, sequence_length in zip(model_result['bs']			,  model_result['ss']	):
										_snake_case		=			model_result['result'][batch_size][sequence_length]
										self.assertIsNotNone(lowercase	)



	def  A							(  self	:    str	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    Any	):


				'''simple docstring'''


				_snake_case		=			'sgugger/tiny-distilbert-classification'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  only_pretrain_model=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    Optional[int]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  torchscript=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	@unittest.skipIf(torch_device == 'cpu'			,  'Cant do half precision'	)
	def  A							(  self	:    Optional[Any]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  fpaa=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    str	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			AutoConfig.from_pretrained(lowercase	)
				# set architectures equal to `None`
				_snake_case		=			None
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase			,  configs=[config]	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    Optional[Any]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_train_result	)
				self.check_results_dict_not_empty(results.memory_train_result	)



	@unittest.skipIf(torch_device == 'cpu'			,  'Can\'t do half precision'	)
	def  A							(  self	:    str	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  fpaa=lowercase			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_train_result	)
				self.check_results_dict_not_empty(results.memory_train_result	)



	def  A							(  self	:    Tuple	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			AutoConfig.from_pretrained(lowercase	)
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase			,  configs=[config]	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    Union[str, Any]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tinier_bart'
				_snake_case		=			AutoConfig.from_pretrained(lowercase	)
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase			,  configs=[config]	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_inference_result	)
				self.check_results_dict_not_empty(results.memory_inference_result	)



	def  A							(  self	:    Dict	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				_snake_case		=			AutoConfig.from_pretrained(lowercase	)
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase			,  configs=[config]	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_train_result	)
				self.check_results_dict_not_empty(results.memory_train_result	)



	def  A							(  self	:    Dict	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tinier_bart'
				_snake_case		=			AutoConfig.from_pretrained(lowercase	)
				_snake_case		=			PyTorchBenchmarkArguments(
				    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  multi_process=lowercase			,  )
				_snake_case		=			PyTorchBenchmark(lowercase			,  configs=[config]	)
				_snake_case		=			benchmark.run()
				self.check_results_dict_not_empty(results.time_train_result	)
				self.check_results_dict_not_empty(results.memory_train_result	)



	def  A							(  self	:    Optional[Any]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'
				with tempfile.TemporaryDirectory() as tmp_dir:
							_snake_case		=			PyTorchBenchmarkArguments(
							    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  save_to_csv=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  inference_time_csv_file=os.path.join(lowercase			,  'inf_time.csv'	)			,  train_memory_csv_file=os.path.join(lowercase			,  'train_mem.csv'	)			,  inference_memory_csv_file=os.path.join(lowercase			,  'inf_mem.csv'	)			,  train_time_csv_file=os.path.join(lowercase			,  'train_time.csv'	)			,  env_info_csv_file=os.path.join(lowercase			,  'env.csv'	)			,  multi_process=lowercase			,  )
							_snake_case		=			PyTorchBenchmark(lowercase	)
							benchmark.run()
							self.assertTrue(Path(os.path.join(lowercase			,  'inf_time.csv'	)	).exists()	)
							self.assertTrue(Path(os.path.join(lowercase			,  'train_time.csv'	)	).exists()	)
							self.assertTrue(Path(os.path.join(lowercase			,  'inf_mem.csv'	)	).exists()	)
							self.assertTrue(Path(os.path.join(lowercase			,  'train_mem.csv'	)	).exists()	)
							self.assertTrue(Path(os.path.join(lowercase			,  'env.csv'	)	).exists()	)





	def  A							(  self	:    Union[str, Any]	):


				'''simple docstring'''


				_snake_case		=			'sshleifer/tiny-gpt2'

				def _check_summary_is_not_empty(lowercase	:    Optional[Any]	):
							self.assertTrue(hasattr(lowercase			,  'sequential'	)	)
							self.assertTrue(hasattr(lowercase			,  'cumulative'	)	)
							self.assertTrue(hasattr(lowercase			,  'current'	)	)
							self.assertTrue(hasattr(lowercase			,  'total'	)	)

				with tempfile.TemporaryDirectory() as tmp_dir:
							_snake_case		=			PyTorchBenchmarkArguments(
							    models=[MODEL_ID]			,  training=lowercase			,  inference=lowercase			,  sequence_lengths=[8]			,  batch_sizes=[1]			,  log_filename=os.path.join(lowercase			,  'log.txt'	)			,  log_print=lowercase			,  trace_memory_line_by_line=lowercase			,  multi_process=lowercase			,  )
							_snake_case		=			PyTorchBenchmark(lowercase	)
							_snake_case		=			benchmark.run()
							_check_summary_is_not_empty(result.inference_summary	)
							_check_summary_is_not_empty(result.train_summary	)
							self.assertTrue(Path(os.path.join(lowercase			,  'log.txt'	)	).exists()	)