Unverified Commit ab2f8d12 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

add hf hub to env version command (#15981)

parent 72983303
......@@ -15,6 +15,8 @@
import platform
from argparse import ArgumentParser
import huggingface_hub
from .. import __version__ as version
from ..file_utils import is_flax_available, is_tf_available, is_torch_available
from . import BaseTransformersCLICommand
......@@ -70,6 +72,7 @@ class EnvironmentCommand(BaseTransformersCLICommand):
"`transformers` version": version,
"Platform": platform.platform(),
"Python version": platform.python_version(),
"Huggingface_hub version": huggingface_hub.__version__,
"PyTorch version (GPU?)": f"{pt_version} ({pt_cuda_available})",
"Tensorflow version (GPU?)": f"{tf_version} ({tf_cuda_available})",
"Flax version (CPU?/GPU?/TPU?)": f"{flax_version} ({jax_backend})",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment