


from argparse import ArgumentParser

from . import BaseTransformersCLICommand






def   _lowercase(					__a				:						Optional[int]     ):
							return DownloadCommand(args.model    ,     args.cache_dir    ,     args.force    ,     args.trust_remote_code     )







class       lowercase_					(lowercase__   ):





				@staticmethod
				def 						__UpperCamelCase    (				lowercase_)  -> Optional[Any]:
											a__      =parser.add_parser('download')
											download_parser.add_argument(
											    '--cache-dir'       ,  type=lowercase_       ,  default=lowercase_       ,  help='Path to location to store the models')
											download_parser.add_argument(
											    '--force'       ,  action='store_true'       ,  help='Force the model to be download even if already in cache-dir')
											download_parser.add_argument(
											    '--trust-remote-code'       ,  action='store_true'       ,  help='Whether or not to allow for custom models defined on the Hub in their own modeling files. Use only if you\'ve reviewed the code as it will execute on your local machine'       ,  )
											download_parser.add_argument('model'       ,  type=lowercase_       ,  help='Name of the model to download')
											download_parser.set_defaults(func=lowercase_)





				def __init__(				self       ,  lowercase_       ,  lowercase_       ,  lowercase_       ,  lowercase_)  -> str:
											a__      =model
											a__      =cache
											a__      =force
											a__      =trust_remote_code





				def 						__UpperCamelCase    (				self)  -> Optional[int]:
											from ..models.auto import AutoModel, AutoTokenizer

											AutoModel.from_pretrained(
											    self._model       ,  cache_dir=self._cache       ,  force_download=self._force       ,  trust_remote_code=self._trust_remote_code)
											AutoTokenizer.from_pretrained(
											    self._model       ,  cache_dir=self._cache       ,  force_download=self._force       ,  trust_remote_code=self._trust_remote_code)





from manim import *




class       lowercase_					(lowercase__   ):





				def 						__UpperCamelCase    (				self)  -> List[Any]:
											a__      =Rectangle(height=0.5       ,  width=0.5)
											a__      =Rectangle(height=0.46       ,  width=0.46).set_stroke(width=0)

											a__      =[mem.copy() for i in range(6)]
											a__      =[mem.copy() for i in range(6)]
											a__      =VGroup(*lowercase_).arrange(lowercase_       ,  buff=0)
											a__      =VGroup(*lowercase_).arrange(lowercase_       ,  buff=0)
											a__      =VGroup(lowercase_       ,  lowercase_).arrange(lowercase_       ,  buff=0)
											a__      =Text('CPU'       ,  font_size=24)
											a__      =Group(lowercase_       ,  lowercase_).arrange(lowercase_       ,  buff=0.5       ,  aligned_edge=lowercase_)
											cpu.move_to([-2.5, -0.5, 0])
											self.add(lowercase_)

											a__      =[mem.copy() for i in range(4)]
											a__      =VGroup(*lowercase_).arrange(lowercase_       ,  buff=0)
											a__      =Text('GPU'       ,  font_size=24)
											a__      =Group(lowercase_       ,  lowercase_).arrange(lowercase_       ,  buff=0.5       ,  aligned_edge=lowercase_)
											gpu.move_to([-1, -1, 0])
											self.add(lowercase_)

											a__      =[mem.copy() for i in range(6)]
											a__      =VGroup(*lowercase_).arrange(lowercase_       ,  buff=0)

											a__      =Text('Model'       ,  font_size=24)
											a__      =Group(lowercase_       ,  lowercase_).arrange(lowercase_       ,  buff=0.5       ,  aligned_edge=lowercase_)
											model.move_to([3, -1.0, 0])
											self.add(lowercase_)

											a__      =[]
											for i, rect in enumerate(lowercase_):
																		rect.set_stroke(lowercase_)
																		# target = fill.copy().set_fill(YELLOW, opacity=0.7)
																		# target.move_to(rect)
																		# self.add(target)

																		a__      =Rectangle(height=0.46 / 4       ,  width=0.46 / 3).set_stroke(width=0.0).set_fill(lowercase_       ,  opacity=0.7)

																		if i == 0:
																									cpu_target.next_to(cpu_left_col_base[0].get_corner(DOWN + LEFT)       ,  buff=0.02       ,  direction=lowercase_)
																									cpu_target.set_x(cpu_target.get_x() + 0.1)
																		elif i == 3:
																									cpu_target.next_to(cpu_targs[0]       ,  direction=lowercase_       ,  buff=0.0)
																		else:
																									cpu_target.next_to(cpu_targs[i - 1]       ,  direction=lowercase_       ,  buff=0.0)
																		self.add(lowercase_)
																		cpu_targs.append(lowercase_)

											a__      =[mem.copy() for i in range(6)]
											a__      =VGroup(*lowercase_).arrange(lowercase_       ,  buff=0)

											a__      =Text('Loaded Checkpoint'       ,  font_size=24)
											a__      =Group(lowercase_       ,  lowercase_).arrange(lowercase_       ,  aligned_edge=lowercase_       ,  buff=0.4)
											checkpoint.move_to([3, 0.5, 0])

											a__      =Square(side_length=2.2)
											key.move_to([-5, 2, 0])

											a__      =MarkupText(
											    F"""<b>Key:</b>\n\n<span fgcolor='{YELLOW}'>●</span> Empty Model"""       ,  font_size=18       ,  )

											key_text.move_to([-5, 2.4, 0])

											self.add(lowercase_       ,  lowercase_)

											a__      =MarkupText(
											    F"""<span fgcolor='{BLUE}'>●</span> Checkpoint"""       ,  font_size=18       ,  )

											blue_text.next_to(lowercase_       ,  DOWN * 2.4       ,  aligned_edge=key_text.get_left())

											a__      =MarkupText(
											    F"""Next, a <i><span fgcolor=\"{BLUE}\">second</span></i> model is loaded into memory,\nwith the weights of a <span fgcolor=\"{BLUE}\">single shard</span>."""       ,  font_size=24       ,  )
											step_a.move_to([2, 2, 0])
											self.play(Write(lowercase_)       ,  Write(lowercase_))

											self.play(Write(lowercase_       ,  run_time=1)       ,  Create(lowercase_       ,  run_time=1))

											a__      =[]
											a__      =[]
											for i, rect in enumerate(lowercase_):
																		a__      =fill.copy().set_fill(lowercase_       ,  opacity=0.7)
																		target.move_to(lowercase_)
																		first_animations.append(GrowFromCenter(lowercase_       ,  run_time=1))

																		a__      =target.copy()
																		cpu_target.generate_target()
																		if i < 5:
																									cpu_target.target.move_to(cpu_left_col_base[i + 1])
																		else:
																									cpu_target.target.move_to(cpu_right_col_base[i - 5])
																		second_animations.append(MoveToTarget(lowercase_       ,  run_time=1.5))

											self.play(*lowercase_)
											self.play(*lowercase_)
											self.wait()

