
    bi                         d dl Z d dlZd dlmZ d dlZd dlmZmZ d dlm	Z	 d dl
mZmZ ddlmZ ddlmZmZmZmZmZ d	d
lmZ d Zedk(  r e        yy)    N)version)default_config_fileload_config_from_file)is_bitsandbytes_available)is_openai_availableis_peft_available   )__version__)is_deepspeed_availableis_diffusers_availableis_liger_kernel_availableis_llm_blender_availableis_vllm_available   )get_git_commit_hashc                     d } t         j                  j                         rRt        t         j                  j	                               D cg c]!  }t         j                  j                  |      # } }nt         j                  j                  j                         rdg} not         j                  j                         rQt        t         j                  j	                               D cg c]!  }t         j                  j                  |      # } }dx}}t        j                  j                  t              rt        t              j                         }t        |t               r;ddj#                  |j%                         D cg c]  \  }}d| d|  c}}      z   n|}t'        d      }i dt)        j(                         dt)        j*                         d	|rt,         d
|d d  nt,        dt/        d      d| dj#                  |       nddt/        d      dt/        d      d|dt/        d      dt/        d      dt1               rt/        d      nddt3               rt/        d      nddt5               rt/        d       ndd!t7               rt/        d"      ndd#t9               rt/        d$      ndd%t;               rt/        d&      ndd't=               rt/        d(      ndd)t?               rt/        d*      ndi}dj#                  |j%                         D cg c]  \  }}d+| d|  c}}      }tA        d,| d       y c c}w c c}w c c}}w c c}}w )-NMPSz	not found
z  - z: trlPlatformzPython versionzTRL version+   zPyTorch versiontorchzaccelerator(s)z, cpuzTransformers versiontransformerszAccelerate version
acceleratezAccelerate configzDatasets versiondatasetszHF Hub versionhuggingface_hubzbitsandbytes versionbitsandbytesznot installedzDeepSpeed version	deepspeedzDiffusers version	diffuserszLiger-Kernel versionliger_kernelzLLM-Blender versionllm_blenderzOpenAI versionopenaizPEFT versionpeftzvLLM versionvllmz- z@
Copy-paste the following information when reporting an issue:

)!r   cudais_availablerangedevice_countget_device_namebackendsmpsxpuospathisfiler   r   to_dict
isinstancedictjoinitemsr   platformpython_versionr
   r   r   r   r   r   r   r   r   r   print)	devicesiaccelerate_configaccelerate_config_strpropvalcommit_hashinfoinfo_strs	            J/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/trl/scripts/env.py	print_envrD   #   s2   Gzz :?

@W@W@Y:Z[Q5::--a0[[				(	(	*'				!9>uyy?U?U?W9XYA599,,Q/YY0;;- 
ww~~)*12EFNNP '. 	tyy@Q@W@W@YZ94Db.Z[[  &e,KH%%'(113 	[+aBQ'89k 	77+	
 	0C$))G, 	 7 	gl3 	2 	GJ/ 	'"34 	;T;V 7\k 	5K5MW[1Sb 	5K5MW[1Sb 	;T;V 7\k 	9Q9Sw}5Yh  	/B/D'(+/!" 	+<+>O#$ 	+<+>O%D* yydjjlKsBtfBse,KLH	OPXzY[
\]U \ Z [8 Ls   
&M&&MM!M
__main__)r/   r7   importlib.metadatar   r   accelerate.commands.configr   r   r   r   transformers.utilsr   r    r
   import_utilsr   r   r   r   r   utilsr   rD   __name__     rC   <module>rO      sI    
  &  Q 2 E   '-^` zK rN   