o ۷iP@sddlZddlZddlmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZdd ZGd d d eZdS) N)ArgumentParser) __version__) is_accelerate_availableis_bitsandbytes_availableis_flax_availableis_google_colabis_peft_availableis_safetensors_availableis_torch_availableis_transformers_availableis_xformers_available)BaseDiffusersCLICommandcCstS)N)EnvironmentCommand)_rL/home/ubuntu/vllm_env/lib/python3.10/site-packages/diffusers/commands/env.pyinfo_command_factory$src@sFeZdZededdfddZdefddZededefd d Z dS) rparserreturnNcCs|d}|jtddS)Nenv)func) add_parser set_defaultsr)rdownload_parserrrrregister_subcommand)s z&EnvironmentCommand.register_subcommandc!Cstj}d}trddl}|j}d}d}tr"ddl}|j}|j}d}d}d} d} trIddl } ddl } ddl } | j}| j}| j} | j j j} d}trUddl}|j}d}traddl}|j}d}trmddl}|j}d}tryddl}|j}d}trddl}|j}t}trdnd}d}tdvrz#tjgdtjtjd}|\}}| d }t!|dkr|"}Wnxt#yYnpwtd kr0zYtjd d gtjtjd}|\}}| d }|$d }|dkr$|t!d 7}|$d|}|||"}|$d}|dkr$|t!d7}|$d|}|d|||"7}Wnt#y/Ynwt%didt&d|d|dt'd|d|dd|d| dd|d| d|d|d|d |d!|d"|d#|d$|d%d&d'd&i} t%d(t%|(| | S))Nz not installedrNAYesNo>LinuxWindows)z nvidia-smiz!--query-gpu=gpu_name,memory.totalz--format=csv,noheader)stdoutstderrzutf-8Darwinsystem_profilerSPDisplaysDataTypezChipset Model: z VRAM (Total):z VRAM: zSIt seems you are running an unusual OS. Could you fill in the accelerator manually?u🤗 Diffusers versionPlatformzRunning on Google Colab?zPython versionzPyTorch version (GPU?)z ()zFlax version (CPU?/GPU?/TPU?)z Jax versionzJaxLib versionzHuggingface_hub versionzTransformers versionzAccelerate versionz PEFT versionzBitsandbytes versionzSafetensors versionzxFormers version AcceleratorzUsing GPU in script?z 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. ))huggingface_hubrr safetensorsr torchcuda is_availablerflaxjaxjaxliblib xla_bridge get_backendplatformr transformersr accelerater peftr bitsandbytesr xformersrsystem subprocessPopenPIPE communicatedecodelenstripFileNotFoundErrorfindprintversionpython_version format_dict)!self hub_versionsafetensors_versionr- pt_versionpt_cuda_availabler. flax_version jax_versionjaxlib_version jax_backendr1r2r3transformers_versionr8accelerate_versionr9 peft_versionr:bitsandbytes_versionr;xformers_versionr< platform_infois_google_colab_str acceleratorspout_strrstartendinforrrrun.s                     zEnvironmentCommand.rundcCsddd|DdS)Nr(cSs g|] \}}d|d|qS)z- z: r).0propvalrrr s z2EnvironmentCommand.format_dict..)joinitems)rbrrrrJszEnvironmentCommand.format_dict) __name__ __module__ __qualname__ staticmethodrrdictrastrrJrrrrr(sr)r7r>argparserr,rrHutilsrrrrr r r r r rrrrrrrs  ,