




"""simple docstring"""






lowercase__     						=						[sum(int(c, 10) ** 2 for c in i.__str__()) for i in range(10_0000)]







def      _snake_case			(				lowercase__		):
				_lowerCamelCase     :			Optional[int]    =    0
				while number:
								# Increased Speed Slightly by checking every 5 digits together.
								sum_of_digits_squared += DIGITS_SQUARED[number % 100000]
								number //= 100000

				return sum_of_digits_squared


# There are 2 Chains made,
# One ends with 89 with the chain member 58 being the one which when declared first,
# there will be the least number of iterations for all the members to be checked.

# The other one ends with 1 and has only one element 1.

# So 58 and 1 are chosen to be declared at the starting.

# Changed dictionary to an array to quicken the solution
lowercase__     						=						[None] * 1000_0000
lowercase__     						=						True
lowercase__     						=						False







def      _snake_case			(				lowercase__		):

				if CHAINS[number - 1] is not None:
								return CHAINS[number - 1]  # type: ignore

				_lowerCamelCase     :			Any    =    chain(next_number(lowercase__		)		)
				_lowerCamelCase     :			Dict    =    number_chain

				while number < 10000000:
								_lowerCamelCase     :			Union[str, Any]    =    number_chain
								number *= 10

				return number_chain







def      _snake_case			(				lowercase__ = 10000000		):
				for i in range(1							,    lowercase__		):
								if CHAINS[i] is None:
												chain(i + 1		)

				return CHAINS[:number].count(lowercase__		)


if __name__ == "__main__":
			import doctest

			doctest.testmod()
			print(F"{solution() = }")


import gc
import unittest

from diffusers import FlaxStableDiffusionInpaintPipeline
from diffusers.utils import is_flax_available, load_image, slow
from diffusers.utils.testing_utils import require_flax


if is_flax_available():
					import jax
					import jax.numpy as jnp
					from flax.jax_utils import replicate
					from flax.training.common_utils import shard







@slow
@require_flax
class  lowercase				(       unittest.TestCase			):




				def     a							(					self       ):
											# clean up the VRAM after each test
											super().tearDown()
											gc.collect()




				def     a							(					self       ):
											snake_case_							=    load_image(
											    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'
											    '/sd2-inpaint/init_image.png'       )
											snake_case_							=    load_image(
											    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd2-inpaint/mask.png'       )

											snake_case_							=    'xvjiarui/stable-diffusion-2-inpainting'
											snake_case_				,    snake_case_							=    FlaxStableDiffusionInpaintPipeline.from_pretrained(snake_case      ,   safety_checker=snake_case       )

											snake_case_							=    'Face of a yellow cat, high resolution, sitting on a park bench'

											snake_case_							=    jax.random.PRNGKey(0       )
											snake_case_							=    50

											snake_case_							=    jax.device_count()
											snake_case_							=    num_samples * [prompt]
											snake_case_							=    num_samples * [init_image]
											snake_case_							=    num_samples * [mask_image]
											snake_case_				,    snake_case_				,    snake_case_							=    pipeline.prepare_inputs(snake_case      ,   snake_case      ,   snake_case       )

											# shard inputs and rng
											snake_case_							=    replicate(snake_case       )
											snake_case_							=    jax.random.split(snake_case      ,   jax.device_count()       )
											snake_case_							=    shard(snake_case       )
											snake_case_							=    shard(snake_case       )
											snake_case_							=    shard(snake_case       )

											snake_case_							=    pipeline(
											    snake_case      ,   snake_case      ,   snake_case      ,   snake_case      ,   snake_case      ,   snake_case      ,   jit=snake_case       )

											snake_case_							=    output.images.reshape(snake_case      ,   512      ,   512      ,   3       )

											snake_case_							=    images[0, 253:256, 253:256, -1]

											snake_case_							=    jnp.asarray(jax.device_get(image_slice.flatten()       )       )
											snake_case_							=    jnp.array(
											    [0.3_61_13_07, 0.37_64_97_36, 0.3_75_74_08, 0.38_21_39_53, 0.39_29_51_67, 0.3_84_16_31, 0.41_55_49_78, 0.4_13_74_75, 0.4_21_70_84]       )
											print(F'''output_slice: {output_slice}'''       )
											assert jnp.abs(output_slice - expected_slice       ).max() < 1e-2




