L i ddlZddlZddlZddlZddlmZddlZddlmZ ddl m Z ddl m Z mZmZmZmZmZmZddlmZd Zd ZGd d eZy) N)ArgumentParser) __version__)is_deepspeed_available)is_accelerate_availableis_flax_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_available)BaseTransformersCLICommandctSN)EnvironmentCommand)_s _/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/transformers/commands/env.pyinfo_command_factoryr&s  c,t|jSr)raccelerate_config_file)argss rdownload_command_factoryr*s d99 ::rc@eZdZedefdZddZdZedZy)rparserc|jd}|jt|jddd|jty)Nenv)funcz--accelerate-config_filezQThe accelerate config file to use for the default values in the launching script.)defaulthelp) add_parser set_defaultsr add_argumentr)rdownload_parsers rregister_subcommandz&EnvironmentCommand.register_subcommand/sT ++E2$$*>$?$$ &d % $$*B$CrNc||_yr)_accelerate_config_file)selfrrs r__init__zEnvironmentCommand.__init__:s '=$rchddl}|j}d}dx}}trddl}ddlm}m}|j}|jtjj|r ||jj}t|tr8dj|jD cgc] \} } d| d| c} } nd|}d} d } d } t!r[ddl}|j} |j$j'} t)}t+}t-}| rd } n|rd } n |rd } n|rd } d}d }t/r+ddl}|j} |j2j5}d}t?r@tAjBtEjF5ddl$}dddj}d}d}d}d }tKr^ddl&}ddl'}ddl(}|j}|j}|j}|jRjTjWjX}tZtYjXtYj\t^j||||| d| d|d|d|d|d||dd}t!r| r#d|d<j$ja|d<nr#d|d<jbja|d<nbr#d|d<jdja|d<n=r;d|d<jfja|d<|jZjh|d<tkdtk|jm||Scc} } w#t6$r(t9|j:j=d}Y.wxYw#1swYxYw)Nrz not installedz not found)default_config_fileload_config_from_file z - :  NACUDAXPUNPUHPUGPUz ()z )z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch version (accelerator?)zTensorflow version (GPU?)zFlax version (CPU?/GPU?/TPU?)z Jax versionzJaxLib versionz/Using distributed or parallel set-up in script?zUsing GPU in script?zGPU typezUsing XPU in script?zXPU typezUsing HPU in script?zHPU typezUsing NPU in script?zNPU typez CANN versionzV Copy-and-paste the text below in your GitHub issue and FILL OUT the two last points. )7 safetensorsrr accelerateaccelerate.commands.configr,r-r(ospathisfileto_dict isinstancedictjoinitemsr torchcuda is_availabler r r r tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblib xla_bridge get_backendplatformversionpython_versionhuggingface_hubget_device_namexpuhpunpucannprint format_dict)r)r9safetensors_versionaccelerate_versionaccelerate_configaccelerate_config_strr:r,r-propval pt_versionpt_cuda_availablept_acceleratorrDpt_xpu_availablept_npu_availablept_hpu_available tf_versiontf_cuda_availabletfdeepspeed_versionrR flax_version jax_versionjaxlib_version jax_backendrSrTrUinfos rrunzEnvironmentCommand.run=s)55,4??1 " $  ]!+!7!7 ++7277>>J];^$9$:V:V$W$_$_$a!/6 ARAXAXAZ[ID#T$r#/[\+,- " %     **J % 7 7 9 57 57 57  !'!!&!!&!!&$     #J Q$&GG$<$<$>! , ! #++BKKM: !  ! ) 5 5 & % (     ++L//K#//N'',,88:CCK'. ))+&557'6'B'B&9%:%7$8$9#:$5#61; B~>Na.P,6Z !/:+,#(99#<#<#>Z !/:+,#(99#<#<#>Z ',}}'9'9^$ hi dt$% }\F" Q$()H)H)O$P! Q ! !s$2M- M3(N'3-N$#N$'N1c dj|jDcgc] \}}d|d|c}}dzScc}}w)Nr.z- r/)rBrC)drhris rrczEnvironmentCommand.format_dicts;yyaggiHsBtfBse,HIDPPHs< )returnN) __name__ __module__ __qualname__ staticmethodrr&r*ryrcrrrr.s@DNDD>pdQQrr)rNrPr<rYargparserr\rrZintegrations.deepspeedrutilsrrr r r r r rrrrrrrrsN  #%;) ;CQ3CQr