
    +hP                         S SK r S SKrS SKJr  S SKrSSKJr  SSKJ	r	J
r
JrJrJrJrJrJrJr  SSKJr  S r " S	 S
\5      rg)    N)ArgumentParser   )__version__)	is_accelerate_availableis_bitsandbytes_availableis_flax_availableis_google_colabis_peft_availableis_safetensors_availableis_torch_availableis_transformers_availableis_xformers_available   )BaseDiffusersCLICommandc                     [        5       $ )N)EnvironmentCommand)_s    P/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/commands/env.pyinfo_command_factoryr   $   s        c                   Z    \ rS rSr\S\SS4S j5       rS\4S jr\S\S\	4S j5       r
S	rg)
r   (   parserreturnNc                 L    U R                  S5      nUR                  [        S9  g )Nenv)func)
add_parserset_defaultsr   )r   download_parsers     r   register_subcommand&EnvironmentCommand.register_subcommand)   s$     ++E2$$*>$?r   c                    [         R                  nSn[        5       (       a  SS KnUR                  nSnSn[	        5       (       a*  SS KnUR                  nUR                  R                  5       nSnSnSn	Sn
[        5       (       a^  SS K	nSS K
nSS KnUR                  nUR                  nUR                  n	UR                  R                  R                  5       R                  n
Sn[!        5       (       a  SS KnUR                  nSn[%        5       (       a  SS KnUR                  nSn[)        5       (       a  SS KnUR                  nSn[-        5       (       a  SS KnUR                  nSn[1        5       (       a  SS KnUR                  n[        R                  " 5       n[5        5       (       a  SOSnSn[        R6                  " 5       S;   az   [8        R:                  " / SQ[8        R<                  [8        R<                  S9nUR?                  5       u  nnURA                  S	5      n[C        U5      S:  a  URE                  5       nGO[        R6                  " 5       S
:X  a   [8        R:                  " SS/[8        R<                  [8        R<                  S9nUR?                  5       u  nnURA                  S	5      nURI                  S5      nUS:w  a  U[C        S5      -  nURI                  SU5      nUUU RE                  5       nURI                  S5      nUS:w  a9  U[C        S5      -  nURI                  SU5      nUSUUU RE                  5       -   -  nO[K        S5        0 S[L        _SU_SU_S[        RN                  " 5       _SU SU S3_SU SU
 S3_SU_SU	_SU_SU_SU_S U_S!U_S"U_S#U_S$U_S%S&_S'S&0En [K        S(5        [K        U RQ                  U 5      5        U $ ! [F         a     Nf = f! [F         a     Nf = f))Nznot installedr   NAYesNo>   LinuxWindows)z
nvidia-smiz!--query-gpu=gpu_name,memory.totalz--format=csv,noheader)stdoutstderrzutf-8Darwinsystem_profilerSPDisplaysDataTypezChipset Model:
zVRAM (Total):z VRAM: zSIt seems you are running an unusual OS. Could you fill in the accelerator manually?u   🤗 Diffusers versionPlatformzRunning on Google Colab?zPython versionzPyTorch version (GPU?)z ()zFlax version (CPU?/GPU?/TPU?)zJax versionzJaxLib versionzHuggingface_hub versionzTransformers versionzAccelerate versionzPEFT versionzBitsandbytes versionzSafetensors versionzxFormers versionAcceleratorzUsing GPU in script?z	<fill in>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_hubr   r   safetensorsr   torchcudais_availabler   flaxjaxjaxliblib
xla_bridgeget_backendplatformr   transformersr   
accelerater
   peftr   bitsandbytesr   xformersr	   system
subprocessPopenPIPEcommunicatedecodelenstripFileNotFoundErrorfindprintversionpython_versionformat_dict)!selfhub_versionsafetensors_versionr4   
pt_versionpt_cuda_availabler5   flax_versionjax_versionjaxlib_versionjax_backendr8   r9   r:   transformers_versionr?   accelerate_versionr@   peft_versionrA   bitsandbytes_versionrB   xformers_versionrC   platform_infois_google_colab_stracceleratorspout_strr   startendinfos!                                    r   runEnvironmentCommand.run.   s@   %11-#%%"-"9"9$
 **J %

 7 7 9&%(++L//K#//N'',,88:CCK.$&&#/#;#; ,"$$!+!7!7&++L.$&&#/#;#; * ""'33 ))+'6'8'8ed?? 44%%`%??%??
  ^^-
!..1w<!#")--/K __(*%%&(<=%??%??
  ^^-
!..1%56B;S!122E!,,tU3C")%"4":":"<K#LL9E{_!55%ll47#y753E3K3K3M'MM gh
$g

 '(;
 h557	

 %B7H6I&K
 ,~R}A-N
 ;
 n
 &{
 #$8
 !"4
 L
 #$8
 "#6
  0
  ;!
" #K#
$ >{%
* 	hidt$%i % . % s&   A7O!  C2O1 !
O.-O.1
O>=O>dc           
          SR                  U R                  5        VVs/ s H  u  pSU SU 3PM     snn5      S-   $ s  snnf )Nr/   z- z: )joinitems)rj   propvals      r   rQ   EnvironmentCommand.format_dict   s=    yyaggiHiBtfBse,iHIDPPHs   >
 )__name__
__module____qualname____firstlineno__staticmethodr   r!   dictrh   strrQ   __static_attributes__rq   r   r   r   r   (   s[    @N @t @ @BT BH Qt Q Q Qr   r   )r>   rE   argparser   r3    r   rO   utilsr   r   r   r	   r
   r   r   r   r   r   r   r   rq   r   r   <module>r}      s@      #  %
 
 
 & LQ0 LQr   