
    wi                     <    d dl Z d dlZd Ze j        d             ZdS )    Nc                    t                      5  t          j        d            | |i |}t          |t          j                  r.|j        j        dk    rt          j                                         |cddd           S # 1 swxY w Y   dS )zWrapper to set seed manually for some functions like dropout
    See: https://github.com/pytorch/pytorch/pull/62315#issuecomment-896143189 for more details.
    *   lazyN)	freeze_rng_statetorchmanual_seed
isinstanceTensordevicetype_lazy	mark_step)opargskwargsoutputs       d/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/torch/testing/_utils.pywrapper_set_seedr      s     
		 	 	"T$V$$fel++ 	$0Bf0L0L K!!###	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   A&BB	Bc            	   #     K   t           j        j                                        5  t           j                                        5  t          j                    } t           j                                        rt           j                                        }d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   	 d V  t           j        j                                        5  t           j                                        5  t           j                                        rt           j        	                    |           t          j	        |            d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S # t           j        j                                        5  t           j                                        5  t           j                                        rt           j        	                    |           t          j	        |            d d d            n# 1 swxY w Y   d d d            w # 1 swxY w Y   w xY w)N)
r   utils_mode_utilsno_dispatch_C_DisableFuncTorchget_rng_statecudais_availableset_rng_state)	rng_statecuda_rng_states     r   r   r      s\      
	 	,	,	.	. 8 80J0J0L0L 8 8'))	:""$$ 	8"Z5577N8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8+ [$0022 	+ 	+EH4N4N4P4P 	+ 	+z&&(( 9
((888	***	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+U[$0022 	+ 	+EH4N4N4P4P 	+ 	+z&&(( 9
((888	***	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+s   B8AB!B8!B%	%B8(B%	)B88B<?B<F +F 
AE(F (E,	,F /E,	0F  FF$I1IAH.	"I.H22I5H26I9II		II	I)
contextlibr   r   contextmanagerr        r   <module>r%      sR            + + + + +r$   