o Zh@sddlZddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddlmZmZmZmZmZmZmZddl mZd d Zd d ZGd ddeZdS)N)ArgumentParser) __version__)is_deepspeed_available)is_accelerate_availableis_flax_availableis_safetensors_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_rH/var/www/auris/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)rZdownload_parserrrrregister_subcommand0s  z&EnvironmentCommand.register_subcommandreturnNcGs ||_dSr)_accelerate_config_file)selfrrrrr__init__;rzEnvironmentCommand.__init__cCsd}tr ddl}|j}ntjddurddl}|jd}d}d}}tr_ddl}ddlm }m }|j}|j dusBt j |rI||j }t|trZddd |Dnd |}d} d } tr}ddl} | j} | j} | j} t} t}d}d }trddl}|j}z|j}Wntyt |j!"d }Ynwd}t#rt$%t&' ddl(}Wdn1swY|j}d}d}d}d }t)rddl*}ddl+}ddl,}|j}|j}|j}|j-j./j0}t1t00t02t3j||||| d | d|d |d|d |d||dd}tra| r/d|d<| j4|d<n2| r>d|d<| j4|d<n#|rMd|d<| j54|d<n| rad|d<| j64|d<| j1j7|d<t8dt8|9||S)Nz not installedr safetensorsz3 but is ignored because of PyTorch version too old.z not found)default_config_fileload_config_from_file cS g|] \}}d|d|qS)z - : r.0propvalrrr U z*EnvironmentCommand.run.. ZNAZGPUz ()z )z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch 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 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. ):rr&r importlibutil find_specr accelerateZaccelerate.commands.configr'r(r#ospathisfileto_dict isinstancedictjoinitemsr torchcudaZ is_availableZxpur r r Z tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubZget_device_nameZhpuZnpuZcannprint format_dict)r$Zsafetensors_versionr&Zaccelerate_versionZaccelerate_configZaccelerate_config_strr8r'r(Z pt_versionZpt_cuda_availablerAZpt_xpu_availableZpt_npu_availableZpt_hpu_availableZ tf_versionZtf_cuda_availabletfZdeepspeed_versionrKZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendrLrMrNinforrrrun>s      zEnvironmentCommand.runcCsddd|DdS)Nr)cSr*)z- r+rr,rrrr0r1z2EnvironmentCommand.format_dict..)r?r@)drrrrVszEnvironmentCommand.format_dict)r"N) __name__ __module__ __qualname__ staticmethodrr!r%rYrVrrrrr/s mr)rGimportlib.utilr5rIr9rQargparserrTrrRZintegrations.deepspeedrutilsrrrr r r r rrrrrrrrs   $