Unverified Commit 1bf4f0da authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Add accelerate and xformers versions to `diffusers-cli env` (#1898)

Add accelerate and xformers to diffusers-cli env
parent f17fae64
......@@ -18,7 +18,7 @@ from argparse import ArgumentParser
import huggingface_hub
from .. import __version__ as version
from ..utils import is_torch_available, is_transformers_available
from ..utils import is_accelerate_available, is_torch_available, is_transformers_available, is_xformers_available
from . import BaseDiffusersCLICommand
......@@ -44,11 +44,23 @@ class EnvironmentCommand(BaseDiffusersCLICommand):
pt_cuda_available = torch.cuda.is_available()
transformers_version = "not installed"
if is_transformers_available:
if is_transformers_available():
import transformers
transformers_version = transformers.__version__
accelerate_version = "not installed"
if is_accelerate_available():
import accelerate
transformers_version = accelerate.__version__
xformers_version = "not installed"
if is_xformers_available():
import xformers
xformers_version = xformers.__version__
info = {
"`diffusers` version": version,
"Platform": platform.platform(),
......@@ -56,6 +68,8 @@ class EnvironmentCommand(BaseDiffusersCLICommand):
"PyTorch version (GPU?)": f"{pt_version} ({pt_cuda_available})",
"Huggingface_hub version": hub_version,
"Transformers version": transformers_version,
"Accelerate version": accelerate_version,
"xFormers version": xformers_version,
"Using GPU in script?": "<fill in>",
"Using distributed or parallel set-up in script?": "<fill in>",
}
......
......@@ -58,6 +58,7 @@ from .import_utils import (
is_transformers_available,
is_transformers_version,
is_unidecode_available,
is_xformers_available,
requires_backends,
)
from .logging import get_logger
......
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