o shp@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)_rW/var/www/html/alpaca_bot/venv/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_subcommand1s  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 } d } trddl} | j} | j} t} t}t}| rd } n| rd } n |rd} n|rd} 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}tru| rCd|d<| j4|d<n2| rRd|d<| j54|d<n#|rad|d<| j64|d<n|rud|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 cS g|] \}}d|d|qS)z - : r.0propvalrrr V z*EnvironmentCommand.run.. NACUDAXPUNPUHPUGPUz ()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 accelerateaccelerate.commands.configr)r*r%ospathisfileto_dict isinstancedictjoinitemsr torchcuda is_availabler r r r tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblib xla_bridge get_backendplatformversionpython_versionhuggingface_hubget_device_namexpuhpunpucannprint format_dict)r&safetensors_versionr(accelerate_versionaccelerate_configaccelerate_config_strr@r)r* pt_versionpt_cuda_availablept_acceleratorrJpt_xpu_availablept_npu_availablept_hpu_available tf_versiontf_cuda_availabletfdeepspeed_versionrX flax_version jax_versionjaxlib_version jax_backendrYrZr[inforrrrun?s     zEnvironmentCommand.runcCsddd|DdS)Nr+cSr,)z- r-rr.rrrr2r3z2EnvironmentCommand.format_dict..)rHrI)drrrriszEnvironmentCommand.format_dict)r$N) __name__ __module__ __qualname__ staticmethodrr#r'r}rirrrrr0s wr)rTimportlib.utilr=rVrBr_argparserrbrr`integrations.deepspeedrutilsrrrr r r r r rrrrrrrrs   (