
    Nj                     B    d dl Z d dlmZ  G d dej                        Zy)    Nc                   H     e Zd Zdedef fdZdej                  fdZ xZ	S )
LayerScalechannelsinitc                     t         |           t        j                  t	        j
                  |f|            | _        y N)super__init__nn	Parametertorchfullscale)selfr   r   	__class__s      o/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/pocket_tts/modules/layer_scale.pyr
   zLayerScale.__init__   s+    \\%**h[$"?@
    xc                      | j                   |z  S r   )r   )r   r   s     r   forwardzLayerScale.forward
   s    zzA~r   )
__name__
__module____qualname__intfloatr
   r   Tensorr   __classcell__)r   s   @r   r   r      s)    A AE A r   r   )r   torch.nnr   Moduler    r   r   <module>r!      s      r   