


'''simple docstring'''
def 			a      (			__a   )  ->	list:



					'''simple docstring'''






					if bit_count < 0:
										raise ValueError('''The given input must be positive'''   )

					# get the generated string sequence
					UpperCamelCase__   :int												= gray_code_sequence_string(_snake_case   )
					#
					# convert them to integers
					for i in range(len(_snake_case   )   ):
										UpperCamelCase__   :Dict												= int(sequence[i] ,					2   )

					return sequence


def 			a      (			__a   )  ->	list:



					'''simple docstring'''






					if bit_count == 0:
										return ["0"]

					if bit_count == 1:
										return ["0", "1"]

					UpperCamelCase__   :int												= 1 << bit_count  # defines the length of the sequence
					# 1<< n is equivalent to 2^n

					# recursive answer will generate answer for n-1 bits
					UpperCamelCase__   :Optional[int]												= gray_code_sequence_string(bit_count - 1   )

					UpperCamelCase__   :Optional[int]												= []

					# append 0 to first half of the smaller sequence generated
					for i in range(seq_len // 2   ):
										UpperCamelCase__   :Optional[int]												= '''0''' + smaller_sequence[i]
										sequence.append(_snake_case   )

					# append 1 to second half ... start from the end of the list
					for i in reversed(range(seq_len // 2   )   ):
										UpperCamelCase__   :Optional[Any]												= '''1''' + smaller_sequence[i]
										sequence.append(_snake_case   )

					return sequence


if __name__ == "__main__":
		import doctest

		doctest.testmod()



"""simple docstring"""
import gc
import unittest

import numpy as np
import torch

from diffusers import StableDiffusionKDiffusionPipeline
from diffusers.utils import slow, torch_device
from diffusers.utils.testing_utils import enable_full_determinism, require_torch_gpu


enable_full_determinism()



@slow
@require_torch_gpu
class 			_UpperCAmelCase (							unittest.TestCase				):






		'''simple docstring'''

		def 				SCREAMING_SNAKE_CASE  (self					):





			'''simple docstring'''

			super().tearDown()
			gc.collect()
			torch.cuda.empty_cache()



		def 				SCREAMING_SNAKE_CASE  (self					):





			'''simple docstring'''

			__snake_case	:						Tuple						=				StableDiffusionKDiffusionPipeline.from_pretrained('''CompVis/stable-diffusion-v1-4'''					)
			__snake_case	:						List[str]						=				sd_pipe.to(a_					)
			sd_pipe.set_progress_bar_config(disable=a_					)

			sd_pipe.set_scheduler('''sample_euler'''					)

			__snake_case	:						Tuple						=				'''A painting of a squirrel eating a burger'''
			__snake_case	:						str						=				torch.manual_seed(0					)
			__snake_case	:						int						=				sd_pipe([prompt]				,       generator=a_				,       guidance_scale=9.0				,       num_inference_steps=20				,       output_type='''np'''					)

			__snake_case	:						List[str]						=				output.images

			__snake_case	:						Tuple						=				image[0, -3:, -3:, -1]

			assert image.shape == (1, 5_12, 5_12, 3)
			__snake_case	:						Dict						=				np.array([0.0447, 0.0492, 0.0468, 0.0408, 0.0383, 0.0408, 0.0354, 0.0380, 0.0339]					)

			assert np.abs(image_slice.flatten() - expected_slice					).max() < 1E-2



		def 				SCREAMING_SNAKE_CASE  (self					):





			'''simple docstring'''

			__snake_case	:						Optional[Any]						=				StableDiffusionKDiffusionPipeline.from_pretrained('''stabilityai/stable-diffusion-2-1-base'''					)
			__snake_case	:						Union[str, Any]						=				sd_pipe.to(a_					)
			sd_pipe.set_progress_bar_config(disable=a_					)

			sd_pipe.set_scheduler('''sample_euler'''					)

			__snake_case	:						str						=				'''A painting of a squirrel eating a burger'''
			__snake_case	:						List[str]						=				torch.manual_seed(0					)
			__snake_case	:						Dict						=				sd_pipe([prompt]				,       generator=a_				,       guidance_scale=9.0				,       num_inference_steps=20				,       output_type='''np'''					)

			__snake_case	:						Any						=				output.images

			__snake_case	:						Tuple						=				image[0, -3:, -3:, -1]

			assert image.shape == (1, 5_12, 5_12, 3)
			__snake_case	:						Tuple						=				np.array([0.1237, 0.1320, 0.1438, 0.1359, 0.1390, 0.1132, 0.1277, 0.1175, 0.1112]					)

			assert np.abs(image_slice.flatten() - expected_slice					).max() < 5E-1



		def 				SCREAMING_SNAKE_CASE  (self					):





			'''simple docstring'''

			__snake_case	:						Any						=				StableDiffusionKDiffusionPipeline.from_pretrained('''stabilityai/stable-diffusion-2-1-base'''					)
			__snake_case	:						Union[str, Any]						=				sd_pipe.to(a_					)
			sd_pipe.set_progress_bar_config(disable=a_					)

			sd_pipe.set_scheduler('''sample_dpmpp_2m'''					)

			__snake_case	:						int						=				'''A painting of a squirrel eating a burger'''
			__snake_case	:						Optional[Any]						=				torch.manual_seed(0					)
			__snake_case	:						Dict						=				sd_pipe(
			    [prompt]				,       generator=a_				,       guidance_scale=7.5				,       num_inference_steps=15				,       output_type='''np'''				,       use_karras_sigmas=a_				,       )

			__snake_case	:						Optional[int]						=				output.images

			__snake_case	:						Any						=				image[0, -3:, -3:, -1]

			assert image.shape == (1, 5_12, 5_12, 3)
			__snake_case	:						Any						=				np.array(
			    [0.1138_1689, 0.1211_2921, 0.138_9457, 0.1254_9606, 0.124_4964, 0.1083_1517, 0.1156_2866, 0.1086_7816, 0.1049_9048]					)

			assert np.abs(image_slice.flatten() - expected_slice					).max() < 1E-2





