
    o9i                     R    d dl mZ  G d de      Z G d de      Z G d de      Zy)	    )Optimizer1Statec                   4     e Zd Z	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )Adagradc                     d|k  st        d|       d|k  st        d|       d|k  st        d|       |dk7  rt        d      |dk7  rt        d      t        | 	  d||d|||||	|
|       y )	N        Invalid learning rate: Invalid weight_decay value: Invalid epsilon value: /Initial accumulator value != 0.0 not supported!Lr Decay != 0.0 not supported!adagradr   r   
ValueErrorsuper__init__selfparamslrlr_decayweight_decayinitial_accumulator_valueeps
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/adagrad.pyr   zAdagrad.__init__	   s     by6rd;<<l".|n=  cz6se<==$+NOOs?=>>	
    
{Gz?r   r   r   绽|=    N   d   T__name__
__module____qualname__r   __classcell__r    s   @r!   r   r      -     "#&
 &
r"   r   c                   4     e Zd Z	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )Adagrad8bitc                     d|k  st        d|       d|k  st        d|       d|k  st        d|       |dk7  rt        d      |dk7  rt        d      |sJ t        | 	  d||d||d	||	|
|       y )
Nr   r   r	   r
   r   r   r   r      r   r   s               r!   r   zAdagrad8bit.__init__3   s     by6rd;<<l".|n=  cz6se<==$+NOOs?=>>z	
r"   )
r$   r   r   r   r%   r3   Nr'   r(   Tr)   r.   s   @r!   r1   r1   2   s-     "#'
 '
r"   r1   c                   4     e Zd Z	 	 	 	 	 	 	 	 	 	 d fd	Z xZS )Adagrad32bitc                     d|k  st        d|       d|k  st        d|       d|k  st        d|       |dk7  rt        d      |dk7  rt        d      t        | 	  d||d||d	||	|
|       y )
Nr   r   r	   r
   r   r   r   r   r&   r   r   s               r!   r   zAdagrad32bit.__init__^   s     by6rd;<<l".|n=  cz6se<==$+NOOs?=>>	
r"   r#   r)   r.   s   @r!   r5   r5   ]   r/   r"   r5   N)bitsandbytes.optim.optimizerr   r   r1   r5    r"   r!   <module>r9      s0   
 9'
o '
T(
/ (
V'
? '
r"   