o i @sddlZddlZddlZddlZddlmZddlZddlmZ ddl m Z ddl m Z mZmZmZmZmZmZddlmZd d Zd d ZGd ddeZdS)N)ArgumentParser) __version__)is_deepspeed_available)is_accelerate_availableis_flax_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_rM/home/ubuntu/.local/lib/python3.10/site-packages/transformers/commands/env.pyinfo_command_factory&srcCs t|jSr)raccelerate_config_file)argsrrrdownload_command_factory* rc@s<eZdZedefddZd ddZdd Zed d ZdS) rparsercCs6|d}|jtd|jdddd|jtddS)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_parserrrrregister_subcommand/s  z&EnvironmentCommand.register_subcommandreturnNcGs ||_dSr)_accelerate_config_file)selfrrrrr__init__:rzEnvironmentCommand.__init__cCsddl}|j}d}d}}trGddl}ddlm}m}|j}|jdus*tj |r1||j }t |t rBddd|Dnd|}d} d } d } trxddl} | j} | j} t} t}t}| rjd } n| rod } n |rtd } n|rxd } d}d }trddl}|j}z|j}Wntyt|jd}Ynwd}trt !t"# ddl$}Wdn1swY|j}d}d}d}d }t%rddl&}ddl'}ddl(}|j}|j}|j}|j)j*+j,}t-t,,t,.t/j||||| d| d|d|d|d|d||dd}tr\| r*d|d<| j0|d<n2| r9d|d<| j10|d<n#|rHd|d<| j20|d<n|r\d|d<| j30|d<| j-j4|d<t5dt5|6||S)Nrz not installedz not found)default_config_fileload_config_from_file cS g|] \}}d|d|qS)z - : r.0propvalrrr O z*EnvironmentCommand.run.. 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%r;safetensors_versionaccelerate_versionaccelerate_configaccelerate_config_strr<r'r( pt_versionpt_cuda_availablept_acceleratorrFpt_xpu_availablept_npu_availablept_hpu_available tf_versiontf_cuda_availabletfdeepspeed_versionrT flax_version jax_versionjaxlib_version jax_backendrUrVrWinforrrrun=s    zEnvironmentCommand.runcCsddd|DdS)Nr)cSr*)z- r+rr,rrrr0r1z2EnvironmentCommand.format_dict..)rDrE)drrrreszEnvironmentCommand.format_dict)r#N) __name__ __module__ __qualname__ staticmethodrr"r&ryrerrrrr.s rr)rPrRr>r[argparserr^rr\integrations.deepspeedrutilsrrrr r r r rrrrrrrrs   $