o  h@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)_rm/var/www/html/construction_image-detection-poc/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_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.. 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 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_availablexpur 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_xpu_availablept_npu_availablept_hpu_available tf_versiontf_cuda_availabletfdeepspeed_versionrT flax_version jax_versionjaxlib_version jax_backendrUrVrWinforrrrun>s      zEnvironmentCommand.runcCsddd|DdS)Nr*cSr+)z- r,rr-rrrr1r2z2EnvironmentCommand.format_dict..)rCrD)drrrrdszEnvironmentCommand.format_dict)r#N) __name__ __module__ __qualname__ staticmethodrr"r&rwrdrrrrr/s mr)rPimportlib.utilr8rRr=r[argparserr^rr\integrations.deepspeedrutilsrrrr r r r rrrrrrrrs   $