o ?"g3@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)_r[/mnt/skqttb/ctump_chatbot/chatbot/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_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 } trxddl} | 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-.j/}t0t//t/1t2j||||| d | d|d |d|d |d||dd}trM| r*d|d<| j3|d<n#| r9d|d<| j43|d<n| rMd|d<| j53|d<| j0j6|d<t7dt7|8||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.. NAGPUz ()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 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. )9rr'r importlibutil find_specr accelerateaccelerate.commands.configr(r)r$ospathisfileto_dict isinstancedictjoinitemsr torchcuda is_availabler r r tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblib xla_bridge get_backendplatformversionpython_versionhuggingface_hubget_device_namehpunpucannprint format_dict)r%safetensors_versionr'accelerate_versionaccelerate_configaccelerate_config_strr;r(r) pt_versionpt_cuda_availablerEpt_npu_availablept_hpu_available tf_versiontf_cuda_availabletfdeepspeed_versionrS flax_version jax_versionjaxlib_version jax_backendrTrUrVinforrrrun>s     zEnvironmentCommand.runcCsddd|DdS)Nr*cSr+)z- r,rr-rrrr1r2z2EnvironmentCommand.format_dict..)rCrD)drrrrcszEnvironmentCommand.format_dict)r#N) __name__ __module__ __qualname__ staticmethodrr"r&rurcrrrrr/s ir)rOimportlib.utilr8rQr=rZargparserr]rr[integrations.deepspeedrutilsrrrr r r r rrrrrrrrs   $