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