o Tia@sddlZddlZddlmZddlZddlmZmZddlm Z ddl m Z m Z ddl mZddlmZmZmZmZmZd d lmZd d Zed krOedSdS)N)version)default_config_fileload_config_from_file)is_bitsandbytes_available)is_openai_availableis_peft_available) __version__)is_deepspeed_availableis_diffusers_availableis_liger_kernel_availableis_llm_blender_availableis_vllm_available)get_git_commit_hashcCsd}tjrddttjD}ntjjrdg}ntjr/ddttjD}d}}tj t r?t t  }t|trRdddd|Dn|}td}id td td |rqtd |dd ntdtdd|durd|nddtddtdd|dtddtddtrtdnddtrtd ndd!trtd"ndd#trtd$ndd%trtd&ndd'trtd(ndd)trtd*ndd+trtd,ndi}dd-d|D}td.|ddS)/NcSg|]}tj|qS)torchcudaget_device_name.0irrC/home/ubuntu/.local/lib/python3.10/site-packages/trl/scripts/env.py ,zprint_env..MPScSrr)rxpurrrrrr0rz not found cS g|] \}}d|d|qS)z - : rrpropvalrrrr9 trlPlatformzPython versionz TRL version+zPyTorch versionrzaccelerator(s)z, cpuzTransformers version transformerszAccelerate version acceleratezAccelerate configzDatasets versiondatasetszHF Hub versionhuggingface_hubzbitsandbytes version bitsandbytesz not installedzDeepSpeed version deepspeedzDiffusers version diffuserszLiger-Kernel version liger_kernelzLLM-Blender version llm_blenderzOpenAI versionopenaiz PEFT versionpeftz vLLM versionvllmcSr)z- r rr!rrrrUr$z@ Copy-paste the following information when reporting an issue: ) rr is_availablerange device_countbackendsmpsrospathisfilerrto_dict isinstancedictjoinitemsrplatformpython_versionr rrr r r r rrrprint)devicesaccelerate_configaccelerate_config_str commit_hashinfoinfo_strrrr print_env)sl          rL__main__)r;rCimportlib.metadatarraccelerate.commands.configrrr*rtransformers.utilsrrr import_utilsr r r r rutilsrrL__name__rrrrs    0