+ j ^RIt^RIHtHt^RIt^RIt^RIHt^RI H t ^RI H t H t ^RIHtHtHtHtHtHtHtHt]!4'd ^RIHuHtR RRlltRR R lltRR R lltR#)N)OptionalUnion)AcceleratorState)CUDA_DISTRIBUTED_TYPES)DistributedTypeRNGType)is_hpu_availableis_mlu_availableis_musa_availableis_neuron_availableis_npu_availableis_sdaa_availableis_torch_xla_availableis_xpu_availablec<V^8dQhR\R\R\/#)seeddevice_specific deterministic)intbool)formats"X/home/luca/screen-checking/.venv/lib/python3.14/site-packages/accelerate/utils/random.py __annotate__r(s!&1&13&1&1d&1cPV'dV\4P, p\P!V4\PPV4\ P !V4\4'd"\ PPV4EM@\4'd"\ PPV4EM\4'd!\ PPV4M\4'd!\ PPV4M\!4'd!\ P"PV4M\%4'd!\ P&PV4MO\)4'd!\ P*PV4M\ P,PV4\/4'd\0P2!V4V'd\ P4!R4R#R#)a Helper function for reproducible behavior to set the seed in `random`, `numpy`, `torch`. Args: seed (`int`): The seed to set. device_specific (`bool`, *optional*, defaults to `False`): Whether to differ the seed on each device slightly with `self.process_index`. deterministic (`bool`, *optional*, defaults to `False`): Whether to use deterministic algorithms where available. Can slow down training. TN)r process_indexrandomrnptorch manual_seedrxpumanual_seed_allr npur mlursdaar musar hpur neuroncudarxm set_rng_stateuse_deterministic_algorithms)rrrs&&&rset_seedr.(sA  "000 KKIINN4 d !!$'    !!$'    !!$'    ""4(    ""4(    !!$'    $$T* ""4(  **40rcpV^8dQhR\\,R\\P,/#)rrng_type generator)rrr Generator)rs"rrrQs/O'O'HW$5O'RWRaRaIbO'rcV\P8Xd\P!4pEMV\P8Xd!\P P4pEMV\P 8XdC\4'gQR4h\P!\P!44pEMAV\P8Xd8\4'gQR4h\PP4pEMV\P8Xd8\4'gQR4h\PP4pEMV\P 8Xd8\#4'gQR4h\P$P4pEM]V\P&8Xd8\)4'gQR4h\P*P4pEMV\P,8Xd7\/4'gQR4h\P0P4pMV\P28Xd7\54'gQR4h\P6P4pM{V\P88Xd7\;4'gQR4h\P<P4pM0V\P>8XdVfQR 4hVPA4p\C4pVPD\FP 8XdcXPI\PJ!44p\PL!V.4\PN!4VPQ4pEMyVPD\R9gVPD\FPT8XgVPD\FPV8XgVPD\FPX8Xg}VPD\FPZ8Xg^VPD\FP\8Xg?VPD\FP^8Xg VPD\FP`8XdMXPIVPb4p\PdPgV^4VPQ4pM?VPD\FPh8Xd!\PdPgX^4V\P8Xd\Pj!X4R#V\P8Xd"\P PkX4R#V\P8Xd"\PPkX4R#V\P8Xd"\PPkX4R#V\P 8Xd"\P$PkX4R#V\P&8Xd"\P*PkX4R#V\P,8Xd"\P0PkX4R#V\P28Xd"\P6PkX4R#V\P88Xd"\P<PkX4R#V\P 8Xd'\Pj!XPm44R#V\P>8XdVPoX4R#R#) z8Can't synchronize XLA seeds as torch_xla is unavailable.z;Can't synchronize NPU seeds on an environment without NPUs.z;Can't synchronize MLU seeds on an environment without MLUs.z=Can't synchronize SDAA seeds on an environment without SDAAs.z=Can't synchronize MUSA seeds on an environment without MUSAs.z;Can't synchronize XPU seeds on an environment without XPUs.z;Can't synchronize HPU seeds on an environment without HPUs.zFCan't synchronize Neuron seeds on an environment without Neuron Cores.Nz)Need a generator to synchronize its seed.)8rTORCHr get_rng_stateCUDAr*XLArtensorr+NPUr r$MLUr r%SDAArr&MUSAr r'XPUrr"HPUr r(NEURONr r) GENERATOR get_staterdistributed_typerto xla_devicecollective_broadcast mark_stepcpur MULTI_MLU MULTI_SDAA MULTI_MUSA MULTI_NPU MULTI_XPU MULTI_HPU MULTI_NEURONdevice distributed broadcast MULTI_CPUr,item set_state)r0r1 rng_statestates&& rsynchronize_rng_staterWQs7== '') W\\ !JJ,,. W[[ %''c)cc'LL!1!1!34 W[[ !!`#``!II++- W[[ !!`#``!II++- W\\ ! ""c$cc"JJ,,. W\\ ! ""c$cc"JJ,,. W[[ !!`#``!II++- W[[ !!`#``!II++- W^^ #"$$n&nn$LL..0 W&& &$Q&QQ$'')   E !4!44LL1   , MMO  "88  ! !_%>%> >  ! !_%?%? ?  ! !_%?%? ?  ! !_%>%> >  ! !_%>%> >  ! !_%>%> >  ! !_%A%A ALL.  ##Iq1MMO   ?#<#< < ##Iq17==  I& W\\ !   + W[[   * W[[   * W\\ !   + W\\ !   + W[[   * W[[   * W^^ # ""9- W[[  )* W&& &I& 'rcV^8dQhR\\\\3,,R\\ P ,/#)r rng_typesr1)listrstrrrr r2)rs"rrrs8FFd5g+>&?FHUZUdUdLeFrcDVFp\\V4VR7K R#))r1N)rWr)rYr1r0s&& rsynchronize_rng_statesr]sgh/9Er)FF)NN)N)rtypingrrnumpyrr rVr constantsr dataclassesrrimportsr r r r r rrrtorch_xla.core.xla_modelcore xla_modelr+r.rWr]rrrgsU" $-1   ))&1RO'dFFr