9iddlZddlZddlZddlmZddlZddlmZ ddl m Z m Z m Z mZmZddlmZdZd ZGd d eZy) N)ArgumentParser) __version__)is_accelerate_availableis_flax_availableis_safetensors_availableis_tf_availableis_torch_available)BaseTransformersCLICommandctSN)EnvironmentCommand)_s c/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/transformers/commands/env.pyinfo_command_factoryr!s  c,t|jSr)raccelerate_config_file)argss rdownload_command_factoryr%s d99 ::rc@eZdZedefdZddZdZedZy)rparserc|jd}|jt|jddd|jty)Nenv)funcz--accelerate-config_filezQThe accelerate config file to use for the default values in the launching script.)defaulthelp) add_parser set_defaultsr add_argumentr)rdownload_parsers rregister_subcommandz&EnvironmentCommand.register_subcommand*sT ++E2$$*>$?$$ &d % $$*B$CrNc||_yr)_accelerate_config_file)selfrrs r__init__zEnvironmentCommand.__init__5s '=$rc`d}trddl}|j}n2tjj dddl}|jd}d}dx}}t rddl}ddlm }m }|j}|jtjj|r ||jj}t!|t"r8dj%|j'D cgc] \} } d| d | c} } nd |}d} d } t)r*ddl} | j} | j,j/} d}d }t1r+ddl}|j} |j4j7}d}d}d}d }tAr^ddl!}ddl"}ddl#}|j}|j}|j}|jHjJjMjN}tPtOjNtOjRtTj|||| d | d|d |d|d |d||ddd}tWdtW|jY||Scc} } w#t8$r(t;|j<j?d }Y+wxYw)Nz not installedr safetensorsz3 but is ignored because of PyTorch version too old.z not found)default_config_fileload_config_from_file z - :  NAGPUz ()z )z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzPyTorch version (GPU?)zTensorflow version (GPU?)zFlax version (CPU?/GPU?/TPU?)z Jax versionzJaxLib versionzUsing 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. )-rr)r importlibutil find_specr accelerateaccelerate.commands.configr*r+r%ospathisfileto_dict isinstancedictjoinitemsr torchcuda is_availabler tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesrflaxjaxjaxliblib xla_bridge get_backendplatformversionpython_versionhuggingface_hubprint format_dict)r&safetensors_versionr)accelerate_versionaccelerate_configaccelerate_config_strr6r*r+propval pt_versionpt_cuda_availabler@ tf_versiontf_cuda_availabletf flax_version jax_versionjaxlib_version jax_backendrJrKrLinfos rrunzEnvironmentCommand.run8s- # % "-"9"9  ^^ % %m 4 @ %0%<%<$==p"q ,4??1 " $  ]!+!7!7 ++7277>>J];^$9$:V:V$W$_$_$a!/6 ARAXAXAZ[ID#T$r#/[\+,- " %     **J % 7 7 9 $     #J Q$&GG$<$<$>! ' % (     ++L//K#//N'',,88:CCK'. ))+&557'6'B'B&9%:%7$8$9#:)3 B7H6I&K,6r?)drZr[s rrUzEnvironmentCommand.format_dicts;yyaggiHsBtfBse,HIDPPHs< )returnN) __name__ __module__ __qualname__ staticmethodrr#r'rfrUrrrr)s@DNDD>SjQQrr)importlib.utilr3r8rPargparserrSrrQutilsrrrr r r rrrrnrrrssC #%) ;fQ3fQr