+ j>^RIt^RIt^RIt^RIt^RIt^RIHtHt^RIt^RI H t ^RI H t ^RI HtHtHt^RIHtHt]!RRR 7tR t]'d^RIHuHt^RItRt]t]P<P?4t RJR lt!R R lt"Rt#Rt$Rt%Rt&Rt'Rt(Rt)Rt*Rt+Rt,Rt-Rt.]RKRl4t/Rt0Rt1Rt2RLRlt3Rt4R t5R!t6R"t7RJR#lt8R$t9R%t:R&t;R'tR*t?R+t@R,tAR-tBR.tCR/tDR0tER1tFR2tGR3tHR4tIR5tJR6tKR7tLR8tMR9tNRMR:ltO]RLR;l4tP]RLR<l4tQ]RLR=l4tR]RLR>l4tS]RLR?l4tTR@tU]RLRAl4tV]RLRBl4tWRCtXRDtYREtZRFt[RGt\RHt]RNRIlt^R# ]dEL'i;i)ON) lru_cachewraps)version)parse)parse_flag_from_envpatch_environment str_to_bool)compare_versionsis_torch_version USE_TORCH_XLAT)defaultFc\PPV4RJpV'd)\PPVfTMT4pR#R# \PPdR#i;i)NTF) importlibutil find_specmetadataPackageNotFoundError)pkg_name metadata_namepackage_exists_s&& Y/home/luca/screen-checking/.venv/lib/python3.14/site-packages/accelerate/utils/imports.py_is_package_availabler2si^^--h7tCN ""++ 8MHS`aA  !!66  s%AA76A7c$V^8dQhR\/#)return)bool)formats"r __annotate__r>s(((c\#N)_torch_distributed_availabler ris_torch_distributed_availabler%>s ''r cz\RR4'd)\PPP 4#R#)>=z2.7.0F)r torch distributeddistributed_c10dis_xccl_availabler$r rr+r+Bs-g&&  11CCEE r c\R4#) import_timerrr$r ris_import_timer_availabler/H  00r c@\R4;'g \RR4#)pynvmlz nvidia-ml-pyr.r$r ris_pynvml_availabler3Ls  * ] ].CHn.]]r c\R4#)amdsmir.r$r ris_amdsmi_availabler6P  **r c\PPRJ;'d\PP 4#r")r(rhipcuda is_availabler$r ris_rocm_availabler<Ts, ==  D ( F FUZZ-D-D-FFr c\R4#)pytestr.r$r ris_pytest_availabler?Xr7r c\RR4#)msampzms-ampr.r$r ris_msamp_availablerB\s ( 33r c\R4#) schedulefreer.r$r ris_schedulefree_availablerE`r0r cR\4'd \RR4#\RR4#)intel_transformer_enginezintel-transformer-enginetransformer_enginetransformer-engine)is_hpu_availablerr$r ris_transformer_engine_availablerKds)$%?A[\\$%9;OPPr cR\RR4'd^RIHpV!4^,#R#)rHrIcheck_mxfp8_supportF)rtransformer_engine.pytorch.fp8rNrMs r%is_transformer_engine_mxfp8_availablerPks%13GHHF"$Q'' r c\R4#) lomo_optimr.r$r ris_lomo_availablerSs  ..r c\RR7;_uu_4\PP4pRRR4V# +'giX#;i)zw Checks if `cuda` is available via an `nvml-based` check which won't trigger the drivers and leave cuda uninitialized. 1)PYTORCH_NVML_BASED_CUDA_CHECKN)rr(r:r;) availables ris_cuda_availablerYws?  = =JJ++-  >  > = s > A cV'dV'dQR4h\'gR#V'd"\PP4R9#V'd"\PP4R8H#R#)z Check if `torch_xla` is available. To train a native pytorch job in an environment with torch xla installed, set the USE_TORCH_XLA to false. z6The check_is_tpu and check_is_gpu cannot both be true.FTPUT)GPUCUDA)_torch_xla_available torch_xlaruntime device_type) check_is_tpu check_is_gpus&&ris_torch_xla_availablerds[ h0hh .     ,,./AA   ,,.%77 r c\R4pV'dA\P!\PPR44p\ VRR4#R#)torchaor'z0.6.1Frrrrrr )rtorchao_versions ris_torchao_availableris@*95N!-- (:(:(B(B9(MNw?? r c\R4#) deepspeedr.r$r ris_deepspeed_availablerl  --r c\RR4#r'z2.4.0r r$r ris_pippy_availablerqs D' **r c\RR7'dV'*#\4'd\PP 4#\ 4'd\P P 4#\4'd\PP 4#\4'd+\PPP^^4#R#)z8Checks if bf16 is supported, optionally ignoring the TPUT)rb) rdrYr(r:is_bf16_supportedis_mlu_availablemluis_xpu_availablexpuis_mps_availablebackendsmpsis_macos_or_newer) ignore_tpus&ris_bf16_availabler}s400~zz++--yy**,,yy**,,~~!!33B:: r c*\4'dR#R#)zChecks if fp16 is supportedFT)is_habana_gaudi1r$r ris_fp16_availablers r c^\4;'g\4;'g \4#)zChecks if fp8 is supported)rBrKrir$r ris_fp8_availablers#   ^ ^#B#D ^ ^H\H^^r c\R4pV'dA\P!\PPR44p\ VRR4#R#) bitsandbytesr'z0.39.0Frgr bnb_versions ris_4bit_bnb_availabler@*>:NmmI$6$6$>$>~$NO  T8<< r c\R4pV'dA\P!\PPR44p\ VRR4#R#)rr'z0.37.2Frgrs ris_8bit_bnb_availablerrr c\R4pV'dEVeA\P!\PPR44p\ VRV4#V#)rr'rg) min_versionrrs& ris_bnb_availablersH*>:N+1mmI$6$6$>$>~$NO  T;??r c^\4'gR#^RIpR\VR\449#)FN multi_backendfeatures)rrgetattrset)bnbs r'is_bitsandbytes_multi_backend_availablers'    gc:su= ==r c\R4#) torchvisionr.r$r ris_torchvision_availablers  //r c\\PPRR44^8Xd\P P R4e_\\PPR44p\VRR4'd!\P P RR4#R#R#R# \d%p\P!RT 24Rp?R #Rp?ii;i) ACCELERATE_USE_MEGATRON_LMFalsemegatronNz megatron-corer'0.8.0z .trainingz)Parse Megatron version failed. Exception:F)r osenvirongetrrrrrrr Exceptionwarningswarn)megatron_versiones ris_megatron_lm_availablers2::>>">HIQN >> # #J / ; #(););)C)CO)T#U #$4dGDD$>>33KLLE <O   I!MN sAB22 C!=CC!c\R4#) transformersr.r$r ris_transformers_availablerr0r c\R4#)datasetsr.r$r ris_datasets_availabler  ,,r c\R4#)peftr.r$r ris_peft_availabler  ((r c\R4#)timmr.r$r ris_timm_availablerrr cP\4'd \RR4#\R4#)tritonz triton-xpu)rvrr$r ris_triton_availablers#$X|<<  **r c\R4pV'dA\P!\PPR44p\ VRR4#R#)aim$>u$EF  S':: r c>\R4;'g \R4#) tensorboard tensorboardXr.r$r ris_tensorboard_availablers  / X X3H3XXr c\R4#)wandbr.r$r ris_wandb_availabler  ))r c\R4#)comet_mlr.r$r ris_comet_ml_availablerrr c\R4#)swanlabr.r$r ris_swanlab_availabler  ++r cN\PR8;'d \R4#)trackio)r )sys version_inforr$r ris_trackio_availablers!   w & K K+@+KKr c\R4#)boto3r.r$r ris_boto3_availabler#rr c@\R4'd \RR4#R#)richACCELERATE_ENABLE_RICHF)rrr$r ris_rich_availabler'sV$$"#;UCC r c\R4#) sagemakerr.r$r ris_sagemaker_availabler-rmr c\R4#)tqdmr.r$r ris_tqdm_availabler1rr c\R4#)clearmlr.r$r ris_clearml_availabler5rr c\R4#)pandasr.r$r ris_pandas_availabler9r7r c\R4#) matplotlibr.r$r ris_matplotlib_availabler=rTr c\R4'dR#\PPR4e#\PP R4pR#R# \PP dR#i;i)mlflowTz mlflow-skinnyF)rrrrrr)rs ris_mlflow_availablerAshX&&~~)5 ""++O>3E3E3R3R3T v vY^YgYgYkYkYtYtYvvr c\PPR4fR#^RIp\ RR7;_uu_4\ P P4pRRR4V# +'giX#;i)zv Checks if `mlu` is available via an `cndev-based` check which won't trigger the drivers and leave mlu uninitialized. torch_mluNFrV)PYTORCH_CNDEV_BASED_MLU_CHECK)rrrrrr(rur;) check_devicerrXs& rrtrtUsZ ~~ ,4  = =II**,  >  > = s A%% A6 cj\PPR4fR#^RIpV'd>\P P 4p\P P4#\\R4;'d\P P4# \dR#i;i)zSChecks if `torch_musa` is installed and potentially if a MUSA is in the environment torch_musaNFmusa) rrrrr(r device_countr; RuntimeErrorhasattr)rrrs& ris_musa_availablerf~~ -5  '')A::**, , 5& ! ? ?ejj&=&=&??  ;B## B21B2c\PPR4fR#^RIpT'd>\ P P4p\ P P4#\\ R4;'d\ P P4# \dR#i;i \dR#i;i)zQChecks if `torch_npu` is installed and potentially if a NPU is in the environment torch_npuNFnpu) rrrrrr(rrr;rr)rrrs& ris_npu_availablerxs~~ ,4  &&(A99))+ + 5% = =UYY%;%;%==   s"B$;B6$ B32B36 CCcj\PPR4fR#^RIpV'd>\P P 4p\P P4#\\R4;'d\P P4# \dR#i;i)zSChecks if `torch_sdaa` is installed and potentially if a SDAA is in the environment torch_sdaaNFsdaa) rrrrr(rrr;rr)rrrs& ris_sdaa_availablerrrc&\PPR4e"\PPR4fR#^RIpV'd ^RIHuHuHp\\ R4;'d\ PP4#)zQChecks if `torch.hpu` is installed and potentially if a HPU is in the environmenthabana_frameworksNzhabana_frameworks.torchFhpu) rrrhabana_frameworks.torch(habana_frameworks.torch.distributed.hcclr(r)hcclrrr;) init_hcclrrs& rrJrJsa   !45= >> # #$= > F"?? 5% = =UYY%;%;%==r c\4'd8^RIHuHuHpVP 4VP P8XdR#R#)rNTF)rJ*habana_frameworks.torch.utils.experimentalr(utils experimental_get_device_type synDeviceTypesynDeviceGaudi)htexps rrrs4BB  ! ! #u':':'I'I I r cD\RR4'dR#V'd>\PP4p\PP 4#\ \R4;'d\PP 4# \ dR#i;i)zr Checks if XPU acceleration is available via stock PyTorch (>=2.7) and potentially if a XPU is in the environment z<=z2.6Frw)r r(rwrr;rr)rrs& rrvrvsye$$  &&(A99))+ + 5% = =UYY%;%;%==  s;B BBcj\PPR4fR#V'dB^RIp\P P 4p\P P4#\\R4;'d\P P4# \dR#i;i) torch_neuronxNFneuron) rrrrr(rrr;rr)rrrs& ris_neuron_availablers~~08   ))+A<<,,. . 5( # C C (A(A(CC  s?B## B21B2c\R4#)dvcliver.r$r ris_dvclive_availablerrr c\R4#) torchdatar.r$r ris_torchdata_availablerrmr c\R4pV'dA\P!\PPR44p\ VRR4#R#)rr'rFrg)rtorchdata_versions r*is_torchdata_stateful_dataloader_availablersB*;7N#MM)*<*<*D*D[*QR 14AA r c0a\S4V3Rl4pV#)z[ A decorator that ensures the decorated function is only called when torchao is available. cJ<\4'g \R4hS!V/VB#)ze`torchao` is not available, please install it before calling this function via `pip install torchao`.)ri ImportError)argskwargsfuncs*,rwrapper!torchao_required..wrappers.#%%w T$V$$r rr$r%sf rtorchao_requiredr)s"  4[%% Nr c0a\S4V3Rl4pV#)z[ A decorator that ensures the decorated function is only called when deepspeed is enabled. c<^RIHp^RIHpVP/8wd,V!4P VP 8wd \R4hS!V/VB#)r)AcceleratorState)DistributedTypez|DeepSpeed is not enabled, please make sure that an `Accelerator` is configured for `deepspeed` before calling this function.)accelerate.stater,accelerate.utils.dataclassesr- _shared_statedistributed_type DEEPSPEED ValueError)r"r#r,r-r$s*, rr%#deepspeed_required..wrapper sS5@  ) )R /4D4F4W4W[j[t[t4t0 T$V$$r r'r(sf rdeepspeed_requiredr5s"  4[ % % Nr c\RR4#rorpr$r ris_weights_only_availabler7s D' **r cl\\PPR44p\ VRV4#)numpyr')rrrrr )r numpy_versions& ris_numpy_availabler;s,),,44W=>M M4 ==r r")FF)F)z1.12)z1.25.0)_rimportlib.metadatarrr functoolsrrr( packagingrpackaging.versionr environmentrrr versionsr r r r^torch_xla.core.xla_modelcore xla_modelxmtorch_xla.runtimer_r!_tpu_availabler)r;r#rr%r+r/r3r6r<r?rBrErKrPrSrYrdrirlrqr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrxrtrrrrJrrvrrrrr)r5r7r;r$r rrHsD & #LL8$OTB  -- # & %00==? ( 1^+G+41Q/  ".+ _ >0 1-))+ Y*-,L* .),+/ w      @ @" > >, @ @"  >  >  > >& D D",.  "(+ >m    sFF  F