(tiPddlZddlZddlmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZdZGd d eZy) N)ArgumentParser) __version__) is_accelerate_availableis_bitsandbytes_availableis_flax_availableis_google_colabis_peft_availableis_safetensors_availableis_torch_availableis_transformers_availableis_xformers_available)BaseDiffusersCLICommandctS)N)EnvironmentCommand)_s e/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/diffusers/commands/env.pyinfo_command_factoryr$s  cLeZdZededdfdZdefdZededefdZ y)rparserreturnNcR|jd}|jty)Nenv)func) add_parser set_defaultsr)rdownload_parsers rregister_subcommandz&EnvironmentCommand.register_subcommand)s$ ++E2$$*>$?rctj}d}trddl}|j}d}d}t r*ddl}|j}|j j}d}d}d} d} tr^ddl } ddl } ddl } | j}| j}| j} | jjjj} d}t!rddl}|j}d}t%rddl}|j}d}t)rddl}|j}d}t-rddl}|j}d}t1rddl}|j}tj}t5rdnd}d}tj6dvr{ t9j:gdt8j<t8j<}|j?\}}|jAd }tC|dkDr|jE}ntj6d k(r t9j:d d gt8j<t8j<}|j?\}}|jAd }|jId }|dk7r|tCd z }|jId|}|||jE}|jId}|dk7r9|tCdz }|jId|}|d|||jEzz }n tKdidtLd|d|dtjNd|d|dd|d| dd|d| d|d|d|d |d!|d"|d#|d$|d%d&d'd&i} tKd(tK|jQ| | S#tF$rYwxYw#tF$rYwxYw))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 bitsandbytesrxformersr system subprocessPopenPIPE communicatedecodelenstripFileNotFoundErrorfindprintversionpython_version format_dict)!self hub_versionsafetensors_versionr2 pt_versionpt_cuda_availabler3 flax_version jax_versionjaxlib_version jax_backendr6r7r8transformers_versionr=accelerate_versionr> peft_versionr?bitsandbytes_versionr@xformers_versionrA platform_infois_google_colab_str acceleratorspout_strrstartendinfos! rrunzEnvironmentCommand.run.s%%11 - # % "-"9"9 $     **J % 7 7 9 & % (     ++L//K#//N'',,88:CCK. $ & #/#;#; , " $ !+!7!7 &   ++L. $ & #/#;#; * " '33  ))+ '6'8ed ??  4 4 %%`%??%??  ^^- !..1w{% * hi dt$% i%  .%  s&A8N..C2N=. N:9N:= O O dc dj|jDcgc] \}}d|d|c}}dzScc}}w)Nr-z- z: )joinitems)rgpropvals rrOzEnvironmentCommand.format_dicts?yyaggiHisBtfBse,iHIDPPHs< ) __name__ __module__ __qualname__ staticmethodrr dictrfstrrOrrrr(s[@N@t@@BTBHQtQQQrr)r<rCargparserr1rrMutilsrrrr r r r r rrrrrsrrrws@#%   & LQ0LQr