widdlZddlZddlmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZdZGd d eZdS) N)ArgumentParser) __version__) is_accelerate_availableis_bitsandbytes_availableis_flax_availableis_google_colab is_notebookis_peft_availableis_safetensors_availableis_torch_availableis_transformers_availableis_xformers_available)BaseDiffusersCLICommandctS)N)EnvironmentCommand)_s f/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/commands/env.pyinfo_command_factoryr%s   cZeZdZededdfdZdefdZededefdZ dS)rparserreturnNcf|d}|tdS)Nenv)func) add_parser set_defaultsr)rdownload_parsers rregister_subcommandz&EnvironmentCommand.register_subcommand*s3 ++E22$$*>$?????rctj}d}tr ddl}|j}d}d}t r$ddl}|j}|j}d}d}d} d} trDddl } ddl } ddl } | j}| j}| j} | j j j} d}t!r ddl}|j}d}t%r ddl}|j}d}t)r ddl}|j}d}t-r ddl}|j}d}t1r ddl}|j}tj}t5rdnd}t7rdnd}d}tjdvr t;jgdt:jt:j}| \}}|!d }tE|dkr|#d z}ng#tH$rYnZwxYwtjd kr/ t;jd d gt:jt:j}| \}}|!d }|%d}|dkr|tEdz }|%d|} ||| #}|%d}|dkrJ|tEdz }|%d|} |d||| #zz }n#tH$rYnwxYwtMdidtNd|d|d|dtj(d|d|dd|d| dd|d| d|d |d!|d"|d#|d$|d%|d&|d'd'd(}!tMd)tM|)|!|!S)*Nz not installedrNAYesNo>LinuxWindows)z nvidia-smiz!--query-gpu=gpu_name,memory.totalz--format=csv,noheader)stdoutstderrzutf-8z VRAMDarwinsystem_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 a notebook?zRunning 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 Acceleratorz )zUsing 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. )*huggingface_hubrr safetensorsr torchcuda is_availablerflaxjaxjaxliblib xla_bridge get_backendplatformr transformersr accelerater peftr bitsandbytesrxformersr r system subprocessPopenPIPE communicatedecodelenstripFileNotFoundErrorfindprintversionpython_version format_dict)"self hub_versionsafetensors_versionr3 pt_versionpt_cuda_availabler4 flax_version jax_versionjaxlib_version jax_backendr7r8r9transformers_versionr>accelerate_versionr? peft_versionr@bitsandbytes_versionrAxformers_versionrB platform_infois_notebook_stris_google_colab_str acceleratorspout_strrstartendinfos" rrunzEnvironmentCommand.run/s-%1 - # % % :    "-"9 $      : LLL*J % 7 7 9 9 & % (    D KKK JJJ MMM+L/K#/N',88::CK. $ & & <    #/#; , " $ $ 8    !+!7 &    , KKK+L. $ & & <    #/#; * " " 4 OOO'3  )++ #.==:%%d'6'8'8Beed ?   4 4 4 %```%?%?  ^^-- !..11w<z2EnvironmentCommand.format_dict..s-HHHs,t,,s,,HHHr)joinitems)ris rrPzEnvironmentCommand.format_dicts0yyHHaggiiHHHIIDPPr) __name__ __module__ __qualname__ staticmethodrr!dictrhstrrPrlrrrr)s@N@t@@@\@ETEEEENQtQQQQ\QQQrr)r=rDargparserr2rrNutilsrrrr r r r r rrrrrrlrrr|s0######%%%%%%                        &%%%%%   OQOQOQOQOQ0OQOQOQOQOQr