a  d@spddlZddlZddlmZddlZddlmZddl m Z m Z m Z m Z ddlmZdd ZGd d d eZdS) N)ArgumentParser) __version__)is_flax_availableis_safetensors_availableis_tf_availableis_torch_available)BaseTransformersCLICommandcCstS)N)EnvironmentCommand)_r b/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/commands/env.pyinfo_command_factorysrc@s2eZdZeedddZddZeddZdS) r )parsercCs|d}|jtddS)Nenv)func) add_parser set_defaultsr)rZdownload_parserr r rregister_subcommands z&EnvironmentCommand.register_subcommandc Csnd}trddl}|j}n$tjddur>ddl}|jd}d}d}trdddl}|j}|j }d}d}t rddl }|j}z|j }Wn"tyt|jd}Yn0d} d} d} d} trddl} ddl}ddl}| j} |j} |j} |jjj} ttttj||d|d|d|d| d| d| | d d d }td t|||S) Nz not installedr safetensorsz3 but is ignored because of PyTorch version too old.ZNAZGPUz ()z ) z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzPyTorch 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. )rrr importlibutil find_specrtorchZcudaZ is_availablerZ tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesrflaxjaxjaxliblibZ xla_bridge get_backendplatformversionpython_versionhuggingface_hubprint format_dict)selfZsafetensors_versionrZ pt_versionZpt_cuda_availablerZ tf_versionZtf_cuda_availabletfZ flax_versionZ jax_versionZjaxlib_versionZ jax_backendr!r"r#infor r rrun$sd   zEnvironmentCommand.runcCsddd|DdS)N cSs g|]\}}d|d|qS)z- z: r ).0propvalr r r hz2EnvironmentCommand.format_dict..)joinitems)dr r rr+fszEnvironmentCommand.format_dictN)__name__ __module__ __qualname__ staticmethodrrr/r+r r r rr s Br )Zimportlib.utilrr&argparserr)rr'utilsrrrrr rr r r r rs