o 0 i @sddlZddlZddlZddlZddlmZddlZddlmZ ddl m Z ddl m Z mZmZmZmZmZmZddlmZd d Zd d ZGd ddeZdS)N)ArgumentParser) __version__)is_deepspeed_available)is_accelerate_availableis_flax_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_re/home/app/PaddleOCR-VL-test/.venv_paddleocr/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_subcommand/s  z&EnvironmentCommand.register_subcommandreturnNcGs ||_dSr)_accelerate_config_file)selfrrrrr__init__:rzEnvironmentCommand.__init__cCsddl}|j}d}d}}trGddl}ddlm}m}|j}|jdus*tj |r1||j }t |t rBddd|Dnd|}d} d } d } trxddl} | j} | j} t} t}t}| rjd } n| rod } n |rtd } n|rxd } d}d }trddl}|j}z|j}Wntyt|jd}Ynwd}trt !t"# ddl$}Wdn1swY|j}d}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}tr\| r*d|d<| j0|d<n2| r9d|d<| j10|d<n#|rHd|d<| j20|d<n|r\d|d<| j30|d<| j-j4|d<t5dt5|6||S)Nrz not installedz not found)default_config_fileload_config_from_file cS g|] \}}d|d|qS)z - : r.0propvalrrr O z*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. )7 safetensorsrr accelerateZaccelerate.commands.configr&r'r#ospathisfileto_dict isinstancedictjoinitemsr torchcudaZ is_availabler r r rZ tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesr contextlibredirect_stdoutioStringIO deepspeedrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubZget_device_nameZxpuZhpuZnpuZcannprint format_dict)r$r5Zsafetensors_versionZaccelerate_versionZaccelerate_configZaccelerate_config_strr6r&r'Z pt_versionZpt_cuda_availableZpt_acceleratorr?Zpt_xpu_availableZpt_npu_availableZpt_hpu_availableZ tf_versionZtf_cuda_availabletfZdeepspeed_versionrIZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendrJrKrLinforrrrun=s    zEnvironmentCommand.runcCsddd|DdS)Nr(cSr))z- r*rr+rrrr/r0z2EnvironmentCommand.format_dict..)r=r>)drrrrTszEnvironmentCommand.format_dict)r"N) __name__ __module__ __qualname__ staticmethodrr!r%rWrTrrrrr.s rr)rErGr7rOargparserrRrrPZintegrations.deepspeedrutilsrrrr r r r rrrrrrrrs   $