o h@sddlZddlZddlZddlmZddlZddlmZ ddl m Z m Z m Z mZmZddlmZdd Zd d ZGd d d eZdS)N)ArgumentParser) __version__)is_accelerate_availableis_flax_availableis_safetensors_availableis_tf_availableis_torch_available)BaseTransformersCLICommandcCstSN)EnvironmentCommand)_rO/var/www/html/ai/venv/lib/python3.10/site-packages/transformers/commands/env.pyinfo_command_factory!srcCs t|jSr )r accelerate_config_file)argsrrrdownload_command_factory% rc@s<eZdZedefddZd ddZdd 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)rdownload_parserrrrregister_subcommand*s  z&EnvironmentCommand.register_subcommandreturnNcGs ||_dSr )_accelerate_config_file)selfrrrrr__init__5rzEnvironmentCommand.__init__cCsd}tr ddl}|j}ntjddurddl}|jd}d}d}}tr_ddl}ddlm }m }|j}|j dusBt j |rI||j }t|trZddd |Dnd |}d} d } trrddl} | j} | j} d} d } trddl}|j} z|j} Wntyt|jd } Ynwd}d}d}d }t rddl!}ddl"}ddl#}|j}|j}|j}|j$j%&j'}t(t''t')t*j|||| d | d| d | d|d |d||ddd}t+dt+|,||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 O z*EnvironmentCommand.run.. NAGPUz ()z )z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzPyTorch version (GPU?)zTensorflow version (GPU?)zFlax version (CPU?/GPU?/TPU?)z Jax versionzJaxLib versionzUsing GPU in script?z/Using distributed or parallel set-up in script?zV 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 tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesrflaxjaxjaxliblib xla_bridge get_backendplatformversionpython_versionhuggingface_hubprint format_dict)r"safetensors_versionr$accelerate_versionaccelerate_configaccelerate_config_strr8r%r& pt_versionpt_cuda_availablerB tf_versiontf_cuda_availabletf flax_version jax_versionjaxlib_version jax_backendrLrMrNinforrrrun8s   zEnvironmentCommand.runcCsddd|DdS)Nr'cSr()z- r)rr*rrrr.r/z2EnvironmentCommand.format_dict..)r@rA)drrrrWszEnvironmentCommand.format_dict)r N) __name__ __module__ __qualname__ staticmethodrrr#rfrWrrrrr )s Ur )importlib.utilr5r:rRargparserrUrrSutilsrrrrr r rrr rrrrs