
    o9i
                     R    d dl mZ  G d de      Z G d de      Z G d de      Zy)	    )Optimizer1Statec                   6     e Zd Z	 	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )RMSpropc                 v    |dk(  rt        d      |rt        d      t        | 	  d||||f||||	|
||       y )Nr   )RMSprop with alpha==0.0 is not supported!"Centered RMSprop is not supported!rmspropNotImplementedErrorsuper__init__)selfparamslralphaepsweight_decaymomentumcentered
optim_bitsargsmin_8bit_sizepercentile_clipping
block_wise	__class__s                d/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/bitsandbytes/optim/rmsprop.pyr   zRMSprop.__init__	   s`     A:%;  %&JKKH	
    ){Gz?Gz?:0yE>r   r   F    N   d   T__name__
__module____qualname__r   __classcell__r   s   @r   r   r      s0     !
 !
r   r   c                   4     e Zd Z	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )RMSprop8bitc                 v    |dk(  rt        d      |rt        d      t        | 	  d||||f||d||	|
|       y )Nr   r   r   r	      r
   r   r   r   r   r   r   r   r   r   r   r   r   r   s               r   r   zRMSprop8bit.__init__.   s`     A:%;  %&JKKH	
r   
r   r   r    r   r   FNr"   r#   Tr$   r)   s   @r   r+   r+   -   s-      
  
r   r+   c                   4     e Zd Z	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )RMSprop32bitc                 v    |dk(  rt        d      |rt        d      t        | 	  d||||f||d||	|
|       y )Nr   r   r   r	   r!   r
   r.   s               r   r   zRMSprop32bit.__init__R   s`     A:%;  %&JKKH	
r   r/   r$   r)   s   @r   r1   r1   Q   s-     !
 !
r   r1   N)bitsandbytes.optim.optimizerr   r   r+   r1    r   r   <module>r5      s0   
 9"
o "
J!
/ !
H"
? "
r   