a hp@sddlZddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZddl 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_availableis_torch_xpu_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_rU/var/www/html/assistant/venv/lib/python3.9/site-packages/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_subcommand1s  z&EnvironmentCommand.register_subcommandN)returncGs ||_dSr)_accelerate_config_file)selfrrrrr__init__<szEnvironmentCommand.__init__cCsd}trddl}|j}n$tjddur>ddl}|jd}d}d}}trddl}ddlm }m }|j}|j dust j |r||j }t|trddd |Dnd |}d} d } d } tr*ddl} | j} | j} t} t}t}| rd } n"| rd } n|r d} n |r*d} d}d }tr|ddl}|j}z|j}Wn$tyzt |j!"d}Yn0d}t#rt$%t&'ddl(}Wdn1s0Y|j}d}d}d}d }t)rddl*}ddl+}ddl,}|j}|j}|j}|j-j./j0}t1t00t02t3j||||| d| d|d|d|d|d||dd}tr| rd|d<| j4|d<nd| rd|d<| j54|d<nF|rd|d<| j64|d<n(|rd|d<| j74|d<| j1j8|d<t9dt9|:||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 Vz*EnvironmentCommand.run.. ZNACUDAZXPUZNPUZHPUZGPUz ()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. );rr&r importlibutil find_specr accelerateZaccelerate.commands.configr'r(r#ospathisfileto_dict isinstancedictjoinitemsr torchcudaZ is_availabler r r r Z tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubZget_device_nameZxpuZhpuZnpuZcannprint format_dict)r$Zsafetensors_versionr&Zaccelerate_versionZaccelerate_configZaccelerate_config_strr8r'r(Z pt_versionZpt_cuda_availableZpt_acceleratorrAZpt_xpu_availableZpt_npu_availableZpt_hpu_availableZ tf_versionZtf_cuda_availabletfZdeepspeed_versionrKZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendrLrMrNinforrrrun?s  ( zEnvironmentCommand.runcCsddd|DdS)Nr)cSs g|]\}}d|d|qS)z- r*rr+rrrr/r0z2EnvironmentCommand.format_dict..)r?r@)drrrrVszEnvironmentCommand.format_dict) __name__ __module__ __qualname__ staticmethodrr!r%rYrVrrrrr0s  wr)rGimportlib.utilr5rIr9rQargparserrTrrRZintegrations.deepspeedrutilsrrrr r r r r rrrrrrrrs   (