



from ....configuration_utils import PretrainedConfig
from ....utils import logging


__snake_case						:Union[str, Any]      		=     logging.get_logger(__name__)

__snake_case						:Optional[int]      		=     {
    '''Visual-Attention-Network/van-base''': (
        '''https://huggingface.co/Visual-Attention-Network/van-base/blob/main/config.json'''
    ),
}

class     _A       (	__UpperCAmelCase    ):
       UpperCamelCase__  :    Optional[int]											=						'''van'''



       def __init__(		self							:     Optional[int]     ,		__SCREAMING_SNAKE_CASE							:     Union[str, Any]=224     ,		__SCREAMING_SNAKE_CASE							:     List[str]=3     ,		__SCREAMING_SNAKE_CASE							:     Any=[7, 3, 3, 3]     ,		__SCREAMING_SNAKE_CASE							:     Optional[Any]=[4, 2, 2, 2]     ,		__SCREAMING_SNAKE_CASE							:     Tuple=[64, 128, 320, 512]     ,		__SCREAMING_SNAKE_CASE							:     Optional[Any]=[3, 3, 12, 3]     ,		__SCREAMING_SNAKE_CASE							:     Dict=[8, 8, 4, 4]     ,		__SCREAMING_SNAKE_CASE							:     Any="gelu"     ,		__SCREAMING_SNAKE_CASE							:     Dict=0.02     ,		__SCREAMING_SNAKE_CASE							:     List[Any]=1E-6     ,		__SCREAMING_SNAKE_CASE							:     Optional[Any]=1E-2     ,		__SCREAMING_SNAKE_CASE							:     Optional[int]=0.0     ,		__SCREAMING_SNAKE_CASE							:     Optional[Any]=0.0     ,		**__SCREAMING_SNAKE_CASE							:     Any     ,		):


           '''simple docstring'''

           super().__init__(**__SCREAMING_SNAKE_CASE)
           __a						       =				image_size
           __a						       =				num_channels
           __a						       =				patch_sizes
           __a						       =				strides
           __a						       =				hidden_sizes
           __a						       =				depths
           __a						       =				mlp_ratios
           __a						       =				hidden_act
           __a						       =				initializer_range
           __a						       =				layer_norm_eps
           __a						       =				layer_scale_init_value
           __a						       =				drop_path_rate
           __a						       =				dropout_rate










import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible

class     _A       (	unittest.TestCase    ):



       def 		_lowerCamelCase			(		self							:     List[Any]):


           '''simple docstring'''

           __a						       =				[
               '''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(__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Optional[Any]):


           '''simple docstring'''

           __a						       =				[
               '''unet/diffusion_pytorch_model.bin''',
               '''unet/diffusion_pytorch_model.safetensors''',
           ]
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Optional[Any]):


           '''simple docstring'''

           __a						       =				[
               '''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(__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Dict):


           '''simple docstring'''

           __a						       =				[
               '''text_encoder/pytorch_model.bin''',
               '''text_encoder/model.safetensors''',
           ]
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE))



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


           '''simple docstring'''

           __a						       =				[
               '''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(__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Tuple):


           '''simple docstring'''

           __a						       =				[
               '''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''',
           ]
           __a						       =				'''fp16'''
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Dict):


           '''simple docstring'''

           __a						       =				[
               '''unet/diffusion_pytorch_model.fp16.bin''',
               '''unet/diffusion_pytorch_model.fp16.safetensors''',
           ]
           __a						       =				'''fp16'''
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Optional[Any]):


           '''simple docstring'''

           __a						       =				[
               '''unet/diffusion_pytorch_model.bin''',
               '''unet/diffusion_pytorch_model.safetensors''',
           ]
           __a						       =				'''fp16'''
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))



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


           '''simple docstring'''

           __a						       =				[
               '''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',
           ]
           __a						       =				'''fp16'''
           self.assertFalse(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     Dict):


           '''simple docstring'''

           __a						       =				[
               '''text_encoder/pytorch_model.fp16.bin''',
               '''text_encoder/model.fp16.safetensors''',
           ]
           __a						       =				'''fp16'''
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))



       def 		_lowerCamelCase			(		self							:     List[str]):


           '''simple docstring'''

           __a						       =				[
               '''text_encoder/pytorch_model.bin''',
               '''text_encoder/model.safetensors''',
           ]
           __a						       =				'''fp16'''
           self.assertTrue(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))





       def 		_lowerCamelCase			(		self							:     List[str]):


           '''simple docstring'''

           __a						       =				[
               '''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''',
           ]
           __a						       =				'''fp16'''
           self.assertFalse(is_safetensors_compatible(__SCREAMING_SNAKE_CASE     ,		variant=__SCREAMING_SNAKE_CASE))





