/usr/local/lib/python3.6/site-packages/transformers/commands/__pycache__
Edit: /usr/local/lib/python3.6/site-packages/transformers/commands/__pycache__/env.cpython-36.pyc (2605B)
3
Eg @ sd d dl Z d dlmZ d dlZddlmZ ddlmZm Z m
Z
ddlmZ dd ZG d
d deZ
dS ) N)ArgumentParser )__version__)is_flax_availableis_tf_availableis_torch_available )BaseTransformersCLICommandc C s t S )N)EnvironmentCommand)_ r C/usr/local/lib/python3.6/site-packages/transformers/commands/env.pyinfo_command_factory s r c @ s2 e Zd ZeedddZdd Zedd ZdS ) r
)parserc C s | j d}|jtd d S )Nenv)func)
add_parserset_defaultsr )r Zdownload_parserr r r
register_subcommand s
z&EnvironmentCommand.register_subcommandc C s, d}d}t r&dd l}|j}|jj }d}d}t rvdd l}|j}y|jj }W n$ t k
rt t
|jjd}Y nX d}d}d} d}
t
rdd l}dd l}dd l}
|j}|j}|
j} |jjj j}
ttj tj tj| d| d| d| d| d|
d| | ddd}td t| j| |S )
Nz
not installedZNAr ZGPUz ()z
)z`transformers` versionPlatformzPython versionzHuggingface_hub versionzPyTorch version (GPU?)zTensorflow version (GPU?)zFlax version (CPU?/GPU?/TPU?)zJax 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.
)r torchr ZcudaZis_availabler Z
tensorflowtestZis_gpu_availableAttributeErrorboolconfigZlist_physical_devicesr flaxjaxjaxliblibZ
xla_bridgeget_backendplatformversionpython_versionhuggingface_hubprintformat_dict)selfZ
pt_versionZpt_cuda_availabler Z
tf_versionZtf_cuda_availabletfZflax_versionZjax_versionZjaxlib_versionZjax_backendr r r infor r r
run# sR
zEnvironmentCommand.runc C s dj dd | j D d S )N
c S s g | ]\}}d | d| qS )z- z: r ).0propvalr r r
\ s z2EnvironmentCommand.format_dict..)joinitems)dr r r
r&