
    o9i
                         d dl mZ  G d de      Z G d de      Z G d de      Z G d d	e      Z G d
 de      Z G d de      Zy)    )Optimizer2Statec                   $     e Zd Z	 	 d fd	Z xZS )AdamWc                 <    t         |   d||||||||	|
||       y )Nadamis_pagedsuper__init__selfparamslrbetasepsweight_decayamsgrad
optim_bitsargsmin_8bit_sizepercentile_clipping
block_wiser	   	__class__s                b/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/bitsandbytes/optim/adamw.pyr   zAdamW.__init__
   s9    &&"eS,
TXZgi|  I  T\  	^    MbP?g?g+?:0yE>{Gz?F    N   d   TF__name__
__module____qualname__r   __classcell__r   s   @r   r   r   	   s    suin^ ^r   r   c                   $     e Zd Z	 	 d fd	Z xZS )	AdamW8bitc                 <    t         |   d|||||d||	|
||       y )Nr      r   r
   r   s                r   r   zAdamW8bit.__init__   s4    &&"eS,4Q^`su  KS  	Ur   r   r%   r*   s   @r   r,   r,          suinU Ur   r,   c                   $     e Zd Z	 	 d fd	Z xZS )
AdamW32bitc                 <    t         |   d|||||d||	|
||       y )Nr   r"   r   r
   r   s                r   r   zAdamW32bit.__init__   s8    &&"eS,DR_at  wA  LT  	Ur   r   r%   r*   s   @r   r1   r1      r/   r   r1   c                   $     e Zd Z	 	 d fd	Z xZS )
PagedAdamWc                 <    t         |   d||||||||	|
|d       y )Nr   Tr   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   s               r   r   zPagedAdamW.__init__   s9    &&"eS,
TXZgi|  I  TX  	Yr   
r   r   r    r!   Fr"   Nr#   r$   Tr%   r*   s   @r   r4   r4      s    suZ^Y Yr   r4   c                   $     e Zd Z	 	 d fd	Z xZS )PagedAdamW8bitc                 <    t         |   d|||||d||	|
|d       y )Nr   r.   Tr   r
   r6   s               r   r   zPagedAdamW8bit.__init__   s4    &&"eS,4Q^`su  KO  	Pr   r7   r%   r*   s   @r   r9   r9      s    suZ^P Pr   r9   c                   $     e Zd Z	 	 d fd	Z xZS )PagedAdamW32bitc                 <    t         |   d|||||d||	|
|d       y )Nr   r"   Tr   r
   r6   s               r   r   zPagedAdamW32bit.__init__$   s8    &&"eS,DR_at  wA  LP  	Qr   r7   r%   r*   s   @r   r<   r<   #   s    suZ^Q Qr   r<   N)bitsandbytes.optim.optimizerr   r   r,   r1   r4   r9   r<    r   r   <module>r@      s^   
 9^O ^
U U
U UY Y
P_ P
Qo Qr   