


import argparse

from diffusers.pipelines.stable_diffusion.convert_from_ckpt import download_controlnet_from_original_ckpt


if __name__ == "__main__":
	A			:						Tuple        =  argparse.ArgumentParser()

	parser.add_argument(
	    '--checkpoint_path', default=None, type=str, required=True, help='Path to the checkpoint to convert.'
	)
	parser.add_argument(
	    '--original_config_file',
	    type=str,
	    required=True,
	    help='The YAML config file corresponding to the original architecture.',
	)
	parser.add_argument(
	    '--num_in_channels',
	    default=None,
	    type=int,
	    help='The number of input channels. If `None` number of input channels will be automatically inferred.',
	)
	parser.add_argument(
	    '--image_size',
	    default=5_1_2,
	    type=int,
	    help=(
	        'The image size that the model was trained on. Use 512 for Stable Diffusion v1.X and Stable Siffusion v2'
	        ' Base. Use 768 for Stable Diffusion v2.'
	    ),
	)
	parser.add_argument(
	    '--extract_ema',
	    action='store_true',
	    help=(
	        'Only relevant for checkpoints that have both EMA and non-EMA weights. Whether to extract the EMA weights'
	        ' or not. Defaults to `False`. Add `--extract_ema` to extract the EMA weights. EMA weights usually yield'
	        ' higher quality images for inference. Non-EMA weights are usually better to continue fine-tuning.'
	    ),
	)
	parser.add_argument(
	    '--upcast_attention',
	    action='store_true',
	    help=(
	        'Whether the attention computation should always be upcasted. This is necessary when running stable'
	        ' diffusion 2.1.'
	    ),
	)
	parser.add_argument(
	    '--from_safetensors',
	    action='store_true',
	    help='If `--checkpoint_path` is in `safetensors` format, load checkpoint with safetensors instead of PyTorch.',
	)
	parser.add_argument(
	    '--to_safetensors',
	    action='store_true',
	    help='Whether to store pipeline in safetensors format or not.',
	)
	parser.add_argument('--dump_path', default=None, type=str, required=True, help='Path to the output model.')
	parser.add_argument('--device', type=str, help='Device to use (e.g. cpu, cuda:0, cuda:1, etc.)')






	def 					UpperCamelCase   (      __magic_name__	:				Union[str, Any]	)  ->    str:
			"""simple docstring"""




			if string == "True":
					return True
			elif string == "False":
					return False
			else:
					raise ValueError(f'''could not parse string as bool {string}'''	)

	parser.add_argument(
	    '--use_linear_projection', help='Override for use linear projection', required=False, type=parse_bool
	)

	parser.add_argument('--cross_attention_dim', help='Override for cross attention_dim', required=False, type=int)

	A			:						str        =  parser.parse_args()

	A			:						Optional[int]        =  download_controlnet_from_original_ckpt(
	    checkpoint_path=args.checkpoint_path,
	    original_config_file=args.original_config_file,
	    image_size=args.image_size,
	    extract_ema=args.extract_ema,
	    num_in_channels=args.num_in_channels,
	    upcast_attention=args.upcast_attention,
	    from_safetensors=args.from_safetensors,
	    device=args.device,
	    use_linear_projection=args.use_linear_projection,
	    cross_attention_dim=args.cross_attention_dim,
	)







	controlnet.save_pretrained(args.dump_path, safe_serialization=args.to_safetensors)



'''simple docstring'''

from math import factorial

snake_case									=    {str(digit): factorial(digit) for digit in range(10)}


def        UpperCAmelCase_ (  lowerCamelCase_			):







 """simple docstring"""


 if not isinstance(lowerCamelCase_      ,			lowerCamelCase_			):
  raise TypeError("Parameter number must be int"			)

 if number < 0:
  raise ValueError("Parameter number must be greater than or equal to 0"			)

 # Converts number in string to iterate on its digits and adds its factorial.
 return sum(DIGIT_FACTORIAL[digit] for digit in str(lowerCamelCase_			)			)


def        UpperCAmelCase_ (  lowerCamelCase_ = 6_0      ,			lowerCamelCase_ = 1_0_0_0_0_0_0			):







 """simple docstring"""


 if not isinstance(lowerCamelCase_      ,			lowerCamelCase_			) or not isinstance(lowerCamelCase_      ,			lowerCamelCase_			):
  raise TypeError("Parameters chain_length and number_limit must be int"			)

 if chain_length <= 0 or number_limit <= 0:
  raise ValueError(
      "Parameters chain_length and number_limit must be greater than 0"			)

 # the counter for the chains with the exact desired length
 lowerCAmelCase__     :   str     		=	0
 # the cached sizes of the previous chains
 lowerCAmelCase__     :   dict[int, int]     		=	{}

 for start_chain_element in range(1      ,			lowerCamelCase_			):
  # The temporary set will contain the elements of the chain
  lowerCAmelCase__     :   Any     		=	set()
  lowerCAmelCase__     :   int     		=	0

  # Stop computing the chain when you find a cached size, a repeating item or the
  # length is greater then the desired one.
  lowerCAmelCase__     :   Dict     		=	start_chain_element
  while (
      chain_element not in chain_sets_lengths
      and chain_element not in chain_set
      and chain_set_length <= chain_length
  ):
   chain_set.add(lowerCamelCase_			)
   chain_set_length += 1
   lowerCAmelCase__     :   Dict     		=	digit_factorial_sum(lowerCamelCase_			)

  if chain_element in chain_sets_lengths:
   chain_set_length += chain_sets_lengths[chain_element]

  lowerCAmelCase__     :   Optional[Any]     		=	chain_set_length

  # If chain contains the exact amount of elements increase the counter
  if chain_set_length == chain_length:
   chains_counter += 1

 return chains_counter


if __name__ == "__main__":
     import doctest

     doctest.testmod()
     print(f'{solution()}')





