o i@sdZddlZddlZddlZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZmZmZmZmZ ddeedBejd d fd dfd d ZdddZded efddZdS)zContains commands to print information about the environment and version. Usage: transformers env transformers version N) Annotated) __version__)is_deepspeed_available)is_accelerate_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_availableaccelerate_config_filezQThe accelerate config file to use for the default values in the launching script.)helpreturnc Csddl}|j}d}d}}trEddl}ddlm}m}|j}|dus)tj |r/|| }t |t r@d dd|Dnd |}d} d } d } trvddl} | j} | j} t} t}t}| rhd } n| rmd } n |rrd } n|rvd} d}trtt ddl}Wdn1swY|j}ttttj||||| d| ddd }tr| rd|d<| j|d<n/| rd|d<| j |d<n!|rd|d<| j!|d<n|rd|d<| j"|d<| j#j$|d<t%dt%t&||S)z(Print information about the environment.rNz not installedz not found)default_config_fileload_config_from_file cS g|] \}}d|d|qS)z - : .0propvalrrY/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/transformers/cli/system.py A zenv.. NACUDAXPUNPUHPUz ()z ) z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch version (accelerator?)z/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. )' safetensorsrr accelerateaccelerate.commands.configrrospathisfileto_dict isinstancedictjoinitemsrtorchcuda is_availabler r rr contextlibredirect_stdoutioStringIO deepspeedplatformpython_versionhuggingface_hubget_device_namexpuhpunpuversioncannprint _format_dict)r r#safetensors_versionaccelerate_versionaccelerate_configaccelerate_config_strr$rr pt_versionpt_cuda_availablept_acceleratorr.pt_xpu_availablept_npu_availablept_hpu_availabledeepspeed_versionr5inforrrenv)s      rMcCs ttdS)zPrint CLI version.N)r?rrrrrr=s r=dcCsddd|DdS)NrcSr)z- rrrrrrrrz _format_dict..)r,r-)rNrrrr@sr@)N)r N)__doc__r1r3r&r6typingrr8typerrintegrations.deepspeedrutilsrrrr r strArgumentrMr=r+r@rrrrs.       Z