U 4Af|@sddlZddlZddlZddlmZddlZddlmZ ddl m Z m Z m Z mZmZmZddlmZdd Zd d ZGd d d eZdS)N)ArgumentParser) __version__)is_accelerate_availableis_flax_availableis_safetensors_availableis_tf_availableis_torch_availableis_torch_npu_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_r=/tmp/pip-unpacked-wheel-zw5xktn0/transformers/commands/env.pyinfo_command_factory"srcCs t|jSr )raccelerate_config_file)argsrrrdownload_command_factory&src@s@eZdZeedddZddddZdd Zed d ZdS) r)parsercCs6|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)rZdownload_parserrrrregister_subcommand+s  z&EnvironmentCommand.register_subcommandN)returncGs ||_dSr )_accelerate_config_file)selfrrrrr__init__6szEnvironmentCommand.__init__cCsNd}trddl}|j}n$tjddk r>ddl}|jd}d}d}}trddl}ddlm }m }|j}|j dk st j |r||j }t|trddd |Dnd |}d} d } trddl} | j} | j} t} d} d }tr>ddl}|j} z|j}Wn&tk r<t|j d }YnXd}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 }tr4| r d|d<| j,|d<n(| r4d|d<| j-,|d<| j)j.|d<t/dt/|0||S)Nz not installedr safetensorsz3 but is ignored because of PyTorch version too old.z not found)default_config_fileload_config_from_file cSs g|]\}}d|d|qS)z - : r.0propvalrrr Psz*EnvironmentCommand.run.. ZNAZGPUz ()z ) z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzPyTorch version (GPU?)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 NPU in script?zNPU typez CANN versionzV Copy-and-paste the text below in your GitHub issue and FILL OUT the two last points. )1rr#r importlibutil find_specr accelerateZaccelerate.commands.configr$r%r ospathisfileZto_dict isinstancedictjoinitemsr torchZcudaZ is_availabler rZ tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubZget_device_nameZnpuZcannprint format_dict)r!Zsafetensors_versionr#Zaccelerate_versionZaccelerate_configZaccelerate_config_strr3r$r%Z pt_versionZpt_cuda_availabler;Zpt_npu_availableZ tf_versionZtf_cuda_availabletfZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendr@rArBinforrrrun9s   zEnvironmentCommand.runcCsddd|DdS)Nr&cSs g|]\}}d|d|qS)z- r'rr(rrrr,sz2EnvironmentCommand.format_dict..)r9r:)drrrrJszEnvironmentCommand.format_dict) __name__ __module__ __qualname__ staticmethodrrr"rMrJrrrrr*s  ]r)importlib.utilr0r4rEargparserrHrrFutilsrrrrr r r rrrrrrrs