;iL2SSKrSSKJrJr SSKrSSKrSSKJr SSK J r SSK J r J r SSKJrJrJrJrJrJrJr \"5(a SSKJs Jr SS \S \S \4S jjrSS \\ S\\R:4SjjrSS\\\ \ 4S\\R:4Sjjr!g)N)OptionalUnion)AcceleratorState)CUDA_DISTRIBUTED_TYPES)DistributedTypeRNGType)is_hpu_availableis_mlu_availableis_musa_availableis_npu_availableis_sdaa_availableis_torch_xla_availableis_xpu_availableseeddevice_specific deterministiccU(aU[5R- n[R"U5 [RRU5 [ R "U5 [5(a![ RRU5 GO [5(a [ RRU5 O[5(a [ RRU5 O[5(a [ RRU5 O}[!5(a [ R"RU5 ON[%5(a [ R&RU5 O[ R(RU5 [+5(a[,R."U5 U(a[ R0"S5 gg)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_allrnpur mlursdaar musar hpucudarxm set_rng_stateuse_deterministic_algorithms)rrrs g/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/accelerate/utils/random.pyset_seedr''s&  "000 KKIINN4 d !!$'    !!$'    !!$'    ""4(    ""4(    !!$' ""4(  **40rng_type generatorc U[R:Xa[R"5nGOpU[R:Xa [R R5nGO[@R :XabWRC[RD"55n[RF"U/5 [RH"5 URK5nGORUR>[L;dUR>[@RN:XdUR>[@RP:XdxUR>[@RR:XdZUR>[@RT:Xd<UR>[@RV:XdUR>[@RX:XaLWRCURZ5n[R\R_US 5 URK5nO>UR>[@R`:Xa [R\R_WS 5 U[R:Xa[Rb"W5 gU[R:Xa [R RcW5 gU[R:Xa [RRcW5 gU[R:Xa [RRcW5 gU[R :Xa [R$RcW5 gU[R&:Xa [R*RcW5 gU[R,:Xa [R0RcW5 gW[R2:Xa [R6RcU5 gU[R :Xa%[Rb"URe55 gU[R8:XaURgU5 gg) Nz8Can'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.z)Need a generator to synchronize its seed.r)4r TORCHr get_rng_stateCUDAr"XLArtensorr#NPUrrMLUr rSDAArrMUSAr r XPUrrHPUr r! GENERATOR get_staterdistributed_typer to xla_devicecollective_broadcast mark_stepcpur MULTI_MLU MULTI_SDAA MULTI_MUSA MULTI_NPU MULTI_XPU MULTI_HPUdevice distributed broadcast MULTI_CPUr$item set_state)r)r* rng_statestates r&synchronize_rng_staterMNs<7== '') 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&& &$Q&QQ$'')   E !4!44LL1   , MMO  "88  ! !_%>%> >  ! !_%?%? ?  ! !_%?%? ?  ! !_%>%> >  ! !_%>%> >  ! !_%>%> >LL.  ##Iq1MMO   ?#<#< < ##Iq17==  I& W\\ !   + W[[   * W[[   * W\\ !   + W\\ !   + W[[   * gkk !  * W[[  )* W&& &I& 'r( rng_typesc<UHn[[U5US9 M g)N)r*)rMr )rNr*r)s r&synchronize_rng_statesrPsgh/9Er()FF)NN)N)"rtypingrrnumpyrrrLr constantsr dataclassesr r importsr r r rrrrtorch_xla.core.xla_modelcore xla_modelr#intboolr' GeneratorrMliststrrPr(r&r_s" $-1))$13$1$1d$1NI'HW$5I'RWRaRaIbI'XFd5g+>&?FHUZUdUdLeFr(