U 9%e@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)_rX/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/commands/env.pyinfo_command_factory!srcCs t|jSr )r accelerate_config_file)argsrrrdownload_command_factory%src@s@eZdZeedddZddddZdd 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)rZdownload_parserrrrregister_subcommand*s  z&EnvironmentCommand.register_subcommandN)returncGs ||_dSr )_accelerate_config_file)selfrrrrr__init__5szEnvironmentCommand.__init__cCsd}trddl}|j}n$tjddk r>ddl}|jd}d}d}}trddl}ddlm }m }|j}|j dk st j |r||j }t|trddd |Dnd |}d} d } trddl} | j} | j} d} d } tr8ddl}|j} z|j} Wn&tk r6t|jd } YnXd}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 cSs g|]\}}d|d|qS)z - : r.0propvalrrr Osz*EnvironmentCommand.run.. ZNAZGPUz ()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 accelerateZaccelerate.commands.configr#r$rospathisfileto_dict isinstancedictjoinitemsr torchcudaZ is_availablerZ tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubprint format_dict)r Zsafetensors_versionr"Zaccelerate_versionZaccelerate_configZaccelerate_config_strr2r#r$Z pt_versionZpt_cuda_availabler;Z tf_versionZtf_cuda_availabletfZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendrArBrCinforrrrun8s  zEnvironmentCommand.runcCsddd|DdS)Nr%cSs g|]\}}d|d|qS)z- r&rr'rrrr+sz2EnvironmentCommand.format_dict..)r9r:)drrrrKszEnvironmentCommand.format_dict) __name__ __module__ __qualname__ staticmethodrrr!rNrKrrrrr )s  Ur )importlib.utilr/r3rFargparserrIrrGutilsrrrrr r rrr rrrrs