Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
1bf4f0da
Unverified
Commit
1bf4f0da
authored
Jan 03, 2023
by
Anton Lozhkov
Committed by
GitHub
Jan 03, 2023
Browse files
Add accelerate and xformers versions to `diffusers-cli env` (#1898)
Add accelerate and xformers to diffusers-cli env
parent
f17fae64
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
17 additions
and
2 deletions
+17
-2
src/diffusers/commands/env.py
src/diffusers/commands/env.py
+16
-2
src/diffusers/utils/__init__.py
src/diffusers/utils/__init__.py
+1
-0
No files found.
src/diffusers/commands/env.py
View file @
1bf4f0da
...
@@ -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>"
,
}
}
...
...
src/diffusers/utils/__init__.py
View file @
1bf4f0da
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment