


import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible



class 							_UpperCamelCase (						unittest.TestCase						):

   '''simple docstring'''







   def   _snake_case    (  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""",
             """unet/diffusion_pytorch_model.safetensors""",
         ]
         self.assertTrue(is_safetensors_compatible(_lowerCamelCase							)							)

   def   _snake_case    (  self  : Dict							):






         '''simple docstring'''



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

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






         '''simple docstring'''



         __lowerCamelCase     :      List[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   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						[
             """text_encoder/pytorch_model.bin""",
             """text_encoder/model.safetensors""",
         ]
         self.assertTrue(is_safetensors_compatible(_lowerCamelCase							)							)

   def   _snake_case    (  self  : Optional[Any]							):






         '''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""",
             # Removed: 'text_encoder/model.safetensors',
             """unet/diffusion_pytorch_model.bin""",
             """unet/diffusion_pytorch_model.safetensors""",
         ]
         self.assertFalse(is_safetensors_compatible(_lowerCamelCase							)							)

   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Union[str, Any]       						=						[
             """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.assertTrue(is_safetensors_compatible(_lowerCamelCase       ,  variant=_lowerCamelCase							)							)

   def   _snake_case    (  self  : List[str]							):






         '''simple docstring'''



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

   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						[
             """unet/diffusion_pytorch_model.bin""",
             """unet/diffusion_pytorch_model.safetensors""",
         ]
         __lowerCamelCase     :      List[Any]       						=						"""fp16"""
         self.assertTrue(is_safetensors_compatible(_lowerCamelCase       ,  variant=_lowerCamelCase							)							)

   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Union[str, Any]       						=						[
             """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     :      List[str]       						=						"""fp16"""
         self.assertFalse(is_safetensors_compatible(_lowerCamelCase       ,  variant=_lowerCamelCase							)							)

   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



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

   def   _snake_case    (  self  : Tuple							):






         '''simple docstring'''



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


   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[Any]       						=						[
             """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     :      Optional[Any]       						=						"""fp16"""
         self.assertFalse(is_safetensors_compatible(_lowerCamelCase       ,  variant=_lowerCamelCase							)							)







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


__UpperCamelCase    :						Optional[Any]     =							{
    'configuration_swinv2': ['SWINV2_PRETRAINED_CONFIG_ARCHIVE_MAP', 'Swinv2Config'],
}


try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    __UpperCamelCase    :						List[Any]     =							[
        'SWINV2_PRETRAINED_MODEL_ARCHIVE_LIST',
        'Swinv2ForImageClassification',
        'Swinv2ForMaskedImageModeling',
        'Swinv2Model',
        'Swinv2PreTrainedModel',
    ]


if TYPE_CHECKING:
    from .configuration_swinva import SWINV2_PRETRAINED_CONFIG_ARCHIVE_MAP, SwinvaConfig

    try:
        if not is_torch_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass
    else:
        from .modeling_swinva import (
            SWINV2_PRETRAINED_MODEL_ARCHIVE_LIST,
            SwinvaForImageClassification,
            SwinvaForMaskedImageModeling,
            SwinvaModel,
            SwinvaPreTrainedModel,
        )


else:
    import sys

    __UpperCamelCase    :						List[Any]     =							_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)



