


"""simple docstring"""



from manim import *







class _lowerCAmelCase							( snake_case_      ):







   def 					lowerCamelCase						(      self		)      ->						Any:

       '''simple docstring'''




       snake_case       : List[str]    =       Rectangle(height=0.5		,      width=0.5		)
       snake_case       : Dict    =       Rectangle(height=0.46		,      width=0.46		).set_stroke(width=0		)

       snake_case       : Tuple    =       [mem.copy() for i in range(6		)]
       snake_case       : Any    =       [mem.copy() for i in range(6		)]
       snake_case       : Tuple    =       VGroup(*UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0		)
       snake_case       : Optional[Any]    =       VGroup(*UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0		)
       snake_case       : Optional[Any]    =       VGroup(UpperCamelCase__		,      UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0		)
       snake_case       : Optional[Any]    =       Text("CPU"		,      font_size=24		)
       snake_case       : Optional[int]    =       Group(UpperCamelCase__		,      UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0.5		,      aligned_edge=UpperCamelCase__		)
       cpu.move_to([-2.5, -0.5, 0]		)
       self.add(UpperCamelCase__		)

       snake_case       : Optional[Any]    =       [mem.copy() for i in range(1		)]
       snake_case       : List[str]    =       VGroup(*UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0		)
       snake_case       : Optional[int]    =       Text("GPU"		,      font_size=24		)
       snake_case       : Tuple    =       Group(UpperCamelCase__		,      UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0.5		,      aligned_edge=UpperCamelCase__		)
       gpu.align_to(UpperCamelCase__		,      UpperCamelCase__		)
       gpu.set_x(gpu.get_x() - 1		)

       self.add(UpperCamelCase__		)

       snake_case       : Any    =       [mem.copy() for i in range(6		)]
       snake_case       : Any    =       VGroup(*UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0		)

       snake_case       : Dict    =       Text("Model"		,      font_size=24		)
       snake_case       : Optional[Any]    =       Group(UpperCamelCase__		,      UpperCamelCase__		).arrange(UpperCamelCase__		,      buff=0.5		,      aligned_edge=UpperCamelCase__		)
       model.move_to([3, -1.0, 0]		)

       self.play(
           Create(UpperCamelCase__		,      run_time=1		)		,      Create(UpperCamelCase__		,      run_time=1		)		,      Create(UpperCamelCase__		,      run_time=1		)		,      )

       snake_case       : Optional[Any]    =       MarkupText(
           F'First, an empty model skeleton is loaded\ninto <span fgcolor=\'{YELLOW}\'>memory</span> without using much RAM.'		,      font_size=24		,      )

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

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

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

       step_a.move_to([2, 2, 0]		)
       self.play(Write(UpperCamelCase__		,      run_time=2.5		)		,      Write(UpperCamelCase__		)		,      Write(UpperCamelCase__		)		)

       self.add(UpperCamelCase__		)

       snake_case       : Optional[Any]    =       []
       snake_case       : Dict    =       []
       snake_case       : Union[str, Any]    =       []
       for i, rect in enumerate(UpperCamelCase__		):
           snake_case       : Any    =       Rectangle(height=0.46		,      width=0.46		).set_stroke(width=0.0		).set_fill(UpperCamelCase__		,      opacity=0.7		)
           cpu_target.move_to(UpperCamelCase__		)
           cpu_target.generate_target()
           snake_case       : Optional[int]    =       0.46 / 4
           snake_case       : Any    =       0.46 / 3

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

           first_animations.append(rect.animate(run_time=0.5		).set_stroke(UpperCamelCase__		)		)
           second_animations.append(MoveToTarget(UpperCamelCase__		,      run_time=1.5		)		)

       self.play(*UpperCamelCase__		)
       self.play(*UpperCamelCase__		)

       self.wait()









"""simple docstring"""



from .imports import is_tqdm_available


if is_tqdm_available():
  from tqdm.auto import tqdm as _tqdm

from ..state import PartialState







def     __lowerCAmelCase		(		lowercase    :    bool = True		,				*lowercase    :    str		,				**lowercase    :    List[Any]					)      ->		List[Any]:


    """simple docstring"""





    if not is_tqdm_available():
        raise ImportError("Accelerate's `tqdm` module requires `tqdm` to be installed. Please run `pip install tqdm`."					)
    snake_case       : Dict    =       False
    if main_process_only:
        snake_case       : int    =       PartialState().local_process_index == 0
    return _tqdm(*lowercase		,				**lowercase		,				disable=lowercase					)





