




import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible


class    lowerCamelCase    (		unittest.TestCase						):






			'''simple docstring'''





			def 							UpperCAmelCase__       ( self			:   int  )						->   Union[str, Any]:
								__magic_name__							:	Tuple    							=							[
								    '''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 							UpperCAmelCase__       ( self			:   List[Any]  )						->   Optional[int]:
								__magic_name__							:	Any    							=							[
								    '''unet/diffusion_pytorch_model.bin''',
								    '''unet/diffusion_pytorch_model.safetensors''',
								]
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Optional[Any]  )						->   Optional[Any]:
								__magic_name__							:	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 							UpperCAmelCase__       ( self			:   List[Any]  )						->   Union[str, Any]:
								__magic_name__							:	Dict    							=							[
								    '''text_encoder/pytorch_model.bin''',
								    '''text_encoder/model.safetensors''',
								]
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Any  )						->   Optional[int]:
								__magic_name__							:	str    							=							[
								    '''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 							UpperCAmelCase__       ( self			:   Union[str, Any]  )						->   int:
								__magic_name__							:	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''',
								]
								__magic_name__							:	List[Any]    							=							'''fp16'''
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Optional[Any]  )						->   Tuple:
								__magic_name__							:	int    							=							[
								    '''unet/diffusion_pytorch_model.fp16.bin''',
								    '''unet/diffusion_pytorch_model.fp16.safetensors''',
								]
								__magic_name__							:	List[Any]    							=							'''fp16'''
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   List[Any]  )						->   Optional[int]:
								__magic_name__							:	Tuple    							=							[
								    '''unet/diffusion_pytorch_model.bin''',
								    '''unet/diffusion_pytorch_model.safetensors''',
								]
								__magic_name__							:	List[str]    							=							'''fp16'''
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Optional[Any]  )						->   Optional[int]:
								__magic_name__							:	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''',
								    # Removed: 'unet/diffusion_pytorch_model.fp16.safetensors',
								]
								__magic_name__							:	Optional[int]    							=							'''fp16'''
								self.assertFalse(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Any  )						->   Optional[Any]:
								__magic_name__							:	Optional[int]    							=							[
								    '''text_encoder/pytorch_model.fp16.bin''',
								    '''text_encoder/model.fp16.safetensors''',
								]
								__magic_name__							:	Dict    							=							'''fp16'''
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )

			def 							UpperCAmelCase__       ( self			:   Optional[int]  )						->   List[Any]:
								__magic_name__							:	Any    							=							[
								    '''text_encoder/pytorch_model.bin''',
								    '''text_encoder/model.safetensors''',
								]
								__magic_name__							:	Dict    							=							'''fp16'''
								self.assertTrue(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )


			def 							UpperCAmelCase__       ( self			:   Optional[int]  )						->   List[str]:
								__magic_name__							:	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''',
								]
								__magic_name__							:	List[Any]    							=							'''fp16'''
								self.assertFalse(is_safetensors_compatible(lowerCamelCase_       ,						variant=lowerCamelCase_  )  )









from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_sentencepiece_available,
    is_speech_available,
    is_tf_available,
    is_torch_available,
)


__lowerCamelCase   :						List[Any]				=			{
    '''configuration_speech_to_text''': ['''SPEECH_TO_TEXT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''Speech2TextConfig'''],
    '''processing_speech_to_text''': ['''Speech2TextProcessor'''],
}

try:
	if not is_sentencepiece_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowerCamelCase   :						Any				=			['''Speech2TextTokenizer''']

try:
	if not is_speech_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowerCamelCase   :						Optional[Any]				=			['''Speech2TextFeatureExtractor''']

try:
	if not is_tf_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowerCamelCase   :						Any				=			[
	    '''TF_SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST''',
	    '''TFSpeech2TextForConditionalGeneration''',
	    '''TFSpeech2TextModel''',
	    '''TFSpeech2TextPreTrainedModel''',
	]

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowerCamelCase   :						Any				=			[
	    '''SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST''',
	    '''Speech2TextForConditionalGeneration''',
	    '''Speech2TextModel''',
	    '''Speech2TextPreTrainedModel''',
	]


if TYPE_CHECKING:
	from .configuration_speech_to_text import SPEECH_TO_TEXT_PRETRAINED_CONFIG_ARCHIVE_MAP, SpeechaTextConfig
	from .processing_speech_to_text import SpeechaTextProcessor

	try:
		if not is_sentencepiece_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .tokenization_speech_to_text import SpeechaTextTokenizer

	try:
		if not is_speech_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .feature_extraction_speech_to_text import SpeechaTextFeatureExtractor

	try:
		if not is_tf_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_tf_speech_to_text import (
		    TF_SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST,
		    TFSpeechaTextForConditionalGeneration,
		    TFSpeechaTextModel,
		    TFSpeechaTextPreTrainedModel,
		)

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_speech_to_text import (
		    SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST,
		    SpeechaTextForConditionalGeneration,
		    SpeechaTextModel,
		    SpeechaTextPreTrainedModel,
		)

else:
	import sys

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



