
    (HJjT                     $    d dl mZ  G d d      Zy)    Nc                   4     e Zd Z fdZed        Zd Z xZS )PipelineMixinc                 Z    t         |           d| _        d| _        d| _        d | _        y )Nr      )super__init__pipeline_rankpipeline_size	start_idxend_idx)self	__class__s    `/Users/ahmed/devFolder/claude-voice/.venv/lib/python3.12/site-packages/mlx_lm/models/pipeline.pyr   zPipelineMixin.__init__   s,        c                 H    | j                   | j                  | j                   S )N)layersr   r   )r   s    r   pipeline_layerszPipelineMixin.pipeline_layers   s    {{4>>DLL99r   c                    |j                         | _        |j                         | _        t	        | j
                        | j                  z  }t	        | j
                        || j                  z  z
  }| j                  |k  r|dz  }| j                  | j                  z
  dz
  |z  | _        | j                  |z   | _        | j
                  d | j                   | _        d g| j                  z  | j
                  d | j                   y )Nr   )rankr	   sizer
   lenr   r   r   )r   grouplayers_per_rankextras       r   pipelinezPipelineMixin.pipeline   s     #ZZ\"ZZ\dkk*d.@.@@DKK ?T5G5G#GG%q O,,t/A/AAAEX~~7kk.DLL1)-(?$dnn%r   )__name__
__module____qualname__r   propertyr   r   __classcell__)r   s   @r   r   r      s"     : :@r   r   )mlx.corecoremxr    r   r   <module>r%      s    @ @r   