


"""simple docstring"""






import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible




class 							A_      (       unittest.TestCase ):


  def 							_lowercase					(			self:						List[str]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Optional[int]							=     [
         "safety_checker/pytorch_model.bin",
         "safety_checker/model.safetensors",
         "vae/diffusion_pytorch_model.bin",
         "vae/diffusion_pytorch_model.safetensors",
         "text_encoder/pytorch_model.bin",
         "text_encoder/model.safetensors",
         "unet/diffusion_pytorch_model.bin",
         "unet/diffusion_pytorch_model.safetensors",
     ]
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase    )    )


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







     '''simple docstring'''







     _lowerCamelCase   :				Any							=     [
         "unet/diffusion_pytorch_model.bin",
         "unet/diffusion_pytorch_model.safetensors",
     ]
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						List[str]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Optional[Any]							=     [
         "safety_checker/pytorch_model.bin",
         "safety_checker/model.safetensors",
         "vae/diffusion_pytorch_model.bin",
         "vae/diffusion_pytorch_model.safetensors",
         "text_encoder/pytorch_model.bin",
         "text_encoder/model.safetensors",
         "unet/diffusion_pytorch_model.bin",
         # Removed: 'unet/diffusion_pytorch_model.safetensors',
     ]
     self.assertFalse(is_safetensors_compatible(__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						int    ):







     '''simple docstring'''







     _lowerCamelCase   :				Tuple							=     [
         "text_encoder/pytorch_model.bin",
         "text_encoder/model.safetensors",
     ]
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Optional[Any]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Union[str, Any]							=     [
         "safety_checker/pytorch_model.bin",
         "safety_checker/model.safetensors",
         "vae/diffusion_pytorch_model.bin",
         "vae/diffusion_pytorch_model.safetensors",
         "text_encoder/pytorch_model.bin",
         # Removed: 'text_encoder/model.safetensors',
         "unet/diffusion_pytorch_model.bin",
         "unet/diffusion_pytorch_model.safetensors",
     ]
     self.assertFalse(is_safetensors_compatible(__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Optional[int]    ):







     '''simple docstring'''







     _lowerCamelCase   :				str							=     [
         "safety_checker/pytorch_model.fp16.bin",
         "safety_checker/model.fp16.safetensors",
         "vae/diffusion_pytorch_model.fp16.bin",
         "vae/diffusion_pytorch_model.fp16.safetensors",
         "text_encoder/pytorch_model.fp16.bin",
         "text_encoder/model.fp16.safetensors",
         "unet/diffusion_pytorch_model.fp16.bin",
         "unet/diffusion_pytorch_model.fp16.safetensors",
     ]
     _lowerCamelCase   :				Tuple							=     "fp16"
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Optional[int]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Any							=     [
         "unet/diffusion_pytorch_model.fp16.bin",
         "unet/diffusion_pytorch_model.fp16.safetensors",
     ]
     _lowerCamelCase   :				str							=     "fp16"
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						List[Any]    ):







     '''simple docstring'''







     _lowerCamelCase   :				str							=     [
         "unet/diffusion_pytorch_model.bin",
         "unet/diffusion_pytorch_model.safetensors",
     ]
     _lowerCamelCase   :				Union[str, Any]							=     "fp16"
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Tuple    ):







     '''simple docstring'''







     _lowerCamelCase   :				Optional[int]							=     [
         "safety_checker/pytorch_model.fp16.bin",
         "safety_checker/model.fp16.safetensors",
         "vae/diffusion_pytorch_model.fp16.bin",
         "vae/diffusion_pytorch_model.fp16.safetensors",
         "text_encoder/pytorch_model.fp16.bin",
         "text_encoder/model.fp16.safetensors",
         "unet/diffusion_pytorch_model.fp16.bin",
         # Removed: 'unet/diffusion_pytorch_model.fp16.safetensors',
     ]
     _lowerCamelCase   :				Union[str, Any]							=     "fp16"
     self.assertFalse(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Any    ):







     '''simple docstring'''







     _lowerCamelCase   :				List[str]							=     [
         "text_encoder/pytorch_model.fp16.bin",
         "text_encoder/model.fp16.safetensors",
     ]
     _lowerCamelCase   :				Dict							=     "fp16"
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )


  def 							_lowercase					(			self:						Optional[Any]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Tuple							=     [
         "text_encoder/pytorch_model.bin",
         "text_encoder/model.safetensors",
     ]
     _lowerCamelCase   :				List[Any]							=     "fp16"
     self.assertTrue(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )






  def 							_lowercase					(			self:						List[str]    ):







     '''simple docstring'''







     _lowerCamelCase   :				Tuple							=     [
         "safety_checker/pytorch_model.fp16.bin",
         "safety_checker/model.fp16.safetensors",
         "vae/diffusion_pytorch_model.fp16.bin",
         "vae/diffusion_pytorch_model.fp16.safetensors",
         "text_encoder/pytorch_model.fp16.bin",
         # 'text_encoder/model.fp16.safetensors',
         "unet/diffusion_pytorch_model.fp16.bin",
         "unet/diffusion_pytorch_model.fp16.safetensors",
     ]
     _lowerCamelCase   :				str							=     "fp16"
     self.assertFalse(is_safetensors_compatible(__lowerCAmelCase							,variant=__lowerCAmelCase    )    )



from typing import TYPE_CHECKING

from ....utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


A				:       Optional[Any]      =  {'configuration_van': ['VAN_PRETRAINED_CONFIG_ARCHIVE_MAP', 'VanConfig']}


try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	A				:       Optional[Any]      =  [
	    'VAN_PRETRAINED_MODEL_ARCHIVE_LIST',
	    'VanForImageClassification',
	    'VanModel',
	    'VanPreTrainedModel',
	]

if TYPE_CHECKING:
	from .configuration_van import VAN_PRETRAINED_CONFIG_ARCHIVE_MAP, VanConfig

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_van import (
		    VAN_PRETRAINED_MODEL_ARCHIVE_LIST,
		    VanForImageClassification,
		    VanModel,
		    VanPreTrainedModel,
		)

else:
	import sys

	A				:       Union[str, Any]      =  _LazyModule(__name__, globals()['__file__'], _import_structure)





