+
    ÆÜ³jö  ã                   óT   € ^ RI Ht  ! R R]4      t ! R R]4      t ! R R]4      tR# )	é    )ÚOptimizer1Statec                   ó6   a a€ ] tR t^t oRV 3R lltRtVtV ;t# )ÚRMSpropc                ó‚   <€ V^ 8X  d   \        R4      hV'       d   \        R4      h\        SV `	  RVVW63VVVV	V
4	       R# )aI  
Base RMSprop optimizer.

Arguments:
    params (`torch.tensor`):
        The input parameters to optimize.
    lr (`float`, defaults to 1e-2):
        The learning rate.
    alpha (`float`, defaults to 0.99):
        The alpha value is the decay rate of the squared gradients of the optimizer.
    eps (`float`, defaults to 1e-8):
        The epsilon value prevents division by zero in the optimizer.
    weight_decay (`float`, defaults to 0.0):
        The weight decay value for the optimizer.
    momentum (`float`, defaults to 0):
        The momentum value speeds up the optimizer by taking bigger steps.
    centered (`bool`, defaults to `False`):
        Whether the gradients are normalized by the variance. If `True`, it can help training at the expense of additional compute.
    optim_bits (`int`, defaults to 32):
        The number of bits of the optimizer state.
    args (`object`, defaults to `None`):
        An object with additional arguments.
    min_8bit_size (`int`, defaults to 4096):
        The minimum number of elements of the parameter tensors for 8-bit optimization.
ú)RMSprop with alpha==0.0 is not supported!ú"Centered RMSprop is not supported!ÚrmspropN©ÚNotImplementedErrorÚsuperÚ__init__)ÚselfÚparamsÚlrÚalphaÚepsÚweight_decayÚmomentumÚcenteredÚ
optim_bitsÚargsÚmin_8bit_sizeÚ	__class__s   &&&&&&&&&&&€Ú[/home/luca/screen-checking/.venv/lib/python3.14/site-packages/bitsandbytes/optim/rmsprop.pyr   ÚRMSprop.__init__	   sS   ø€ ðL �AŒ:Ü%Ð&QÓRÐRßÜ%Ð&JÓKÐKÜ‰ÑØØØØÐØØØØØö
	
ó    © )	ç{®Gáz„?ç®Gáz®ï?ç:Œ0âŽyE>r   r   Fé    Né   ©Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r   Ú__classdict__s   @@r   r   r      ó   ù‡ € ÷4
÷ 4
r   r   c                   ó6   a a€ ] tR t^@t oRV 3R lltRtVtV ;t# )ÚRMSprop8bitc
                ó‚   <€ V^ 8X  d   \        R4      hV'       d   \        R4      h\        S
V `	  RVVW63VV^VV	4	       R# )aï  
8-bit RMSprop optimizer.

Arguments:
    params (`torch.tensor`):
        The input parameters to optimize.
    lr (`float`, defaults to 1e-2):
        The learning rate.
    alpha (`float`, defaults to 0.99):
        The alpha value is the decay rate of the squared gradients of the optimizer.
    eps (`float`, defaults to 1e-8):
        The epsilon value prevents division by zero in the optimizer.
    weight_decay (`float`, defaults to 0.0):
        The weight decay value for the optimizer.
    momentum (`float`, defaults to 0):
        The momentum value speeds up the optimizer by taking bigger steps.
    centered (`bool`, defaults to `False`):
        Whether the gradients are normalized by the variance. If `True`, it can help training at the expense of additional compute.
    args (`object`, defaults to `None`):
        An object with additional arguments.
    min_8bit_size (`int`, defaults to 4096):
        The minimum number of elements of the parameter tensors for 8-bit optimization.
r   r   r	   Nr
   ©r   r   r   r   r   r   r   r   r   r   r   s   &&&&&&&&&&€r   r   ÚRMSprop8bit.__init__A   sS   ø€ ðF �AŒ:Ü%Ð&QÓRÐRßÜ%Ð&JÓKÐKÜ‰ÑØØØØÐØØØØØö
	
r   r   ©r   r   r    r   r   FNr"   r#   r+   s   @@r   r/   r/   @   s   ù‡ € ÷1
÷ 1
r   r/   c                   ó6   a a€ ] tR t^ut oRV 3R lltRtVtV ;t# )ÚRMSprop32bitc
                ó‚   <€ V^ 8X  d   \        R4      hV'       d   \        R4      h\        S
V `	  RVVW63VV^ VV	4	       R# )aK  
32-bit RMSprop optimizer.

Arguments:
    params (`torch.tensor`):
        The input parameters to optimize.
    lr (`float`, defaults to 1e-2):
        The learning rate.
    alpha (`float`, defaults to 0.99):
        The alpha value is the decay rate of the squared gradients of the optimizer.
    eps (`float`, defaults to 1e-8):
        The epsilon value prevents division by zero in the optimizer.
    weight_decay (`float`, defaults to 0.0):
        The weight decay value for the optimizer.
    momentum (`float`, defaults to 0):
        The momentum value speeds up the optimizer by taking bigger steps.
    centered (`bool`, defaults to `False`):
        Whether the gradients are normalized by the variance. If `True`, it can help training at the expense of additional compute.
    optim_bits (`int`, defaults to 32):
        The number of bits of the optimizer state.
    args (`object`, defaults to `None`):
        An object with additional arguments.
    min_8bit_size (`int`, defaults to 4096):
        The minimum number of elements of the parameter tensors for 8-bit optimization.
r   r   r	   Nr
   r1   s   &&&&&&&&&&€r   r   ÚRMSprop32bit.__init__v   sS   ø€ ðL �AŒ:Ü%Ð&QÓRÐRßÜ%Ð&JÓKÐKÜ‰ÑØØØØÐØØØØØö
	
r   r   r3   r#   r+   s   @@r   r5   r5   u   r-   r   r5   N)Úbitsandbytes.optim.optimizerr   r   r/   r5   r   r   r   Ú<module>r9      s0   ðõ
 9ô5
ˆoô 5
ôp2
�/ô 2
ôj5
�?ö 5
r   