


lowercase__							:							List[Any]									=			"\n# Transformers installation\n! pip install transformers datasets\n# To install from source instead of the last release, comment the command above and uncomment the following one.\n# ! pip install git+https://github.com/huggingface/transformers.git\n"

lowercase__							:							Union[str, Any]									=			[{"type": "code", "content": INSTALL_CONTENT}]
lowercase__							:							Optional[int]									=			{
    "{processor_class}": "FakeProcessorClass",
    "{model_class}": "FakeModelClass",
    "{object_class}": "FakeObjectClass",
}






import unittest

from diffusers import FlaxAutoencoderKL
from diffusers.utils import is_flax_available
from diffusers.utils.testing_utils import require_flax

from .test_modeling_common_flax import FlaxModelTesterMixin


if is_flax_available():
				import jax






@require_flax
class        SCREAMING_SNAKE_CASE__    (							SCREAMING_SNAKE_CASE_   ,		unittest.TestCase    ):

							"""simple docstring"""






							_snake_case     		=  FlaxAutoencoderKL



							@property
							def     A__		(  self			)->   int:





													'''simple docstring'''






													__UpperCamelCase       =			4
													__UpperCamelCase       =			3
													__UpperCamelCase       =			(32, 32)

													__UpperCamelCase       =			jax.random.PRNGKey(0			)
													__UpperCamelCase       =			jax.random.uniform(SCREAMING_SNAKE_CASE_     ,	((batch_size, num_channels) + sizes)			)

													return {"sample": image, "prng_key": prng_key}



							def     A__		(  self			)->   List[Any]:





													'''simple docstring'''






													__UpperCamelCase       =			{
													    '''block_out_channels''': [32, 64],
													    '''in_channels''': 3,
													    '''out_channels''': 3,
													    '''down_block_types''': ['''DownEncoderBlock2D''', '''DownEncoderBlock2D'''],
													    '''up_block_types''': ['''UpDecoderBlock2D''', '''UpDecoderBlock2D'''],
													    '''latent_channels''': 4,
													}
													__UpperCamelCase       =			self.dummy_input
													return init_dict, inputs_dict




