From 6d6d19d7fc8484eb59ce779eca78760495a9ece5 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Thu, 23 May 2024 06:02:02 +0000 Subject: [PATCH 1/2] update --- src/diffusers/commands/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/commands/env.py b/src/diffusers/commands/env.py index 024b5e6ec786..733d476a4360 100644 --- a/src/diffusers/commands/env.py +++ b/src/diffusers/commands/env.py @@ -152,7 +152,7 @@ def run(self) -> dict: info = { "🤗 Diffusers version": version, - "Platform": f"{platform.freedesktop_os_release().get('PRETTY_NAME', None)} - {platform.platform()}", + "Platform": platform.platform(), "Running on a notebook?": is_notebook_str, "Running on Google Colab?": is_google_colab_str, "Python version": platform.python_version(), From c83683fdd15a1707bdda70e2c04133b1665e3664 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Fri, 24 May 2024 07:38:13 +0000 Subject: [PATCH 2/2] update --- src/diffusers/commands/env.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/diffusers/commands/env.py b/src/diffusers/commands/env.py index 733d476a4360..a828216670ce 100644 --- a/src/diffusers/commands/env.py +++ b/src/diffusers/commands/env.py @@ -31,6 +31,7 @@ is_transformers_available, is_xformers_available, ) +from ..utils.testing_utils import get_python_version from . import BaseDiffusersCLICommand @@ -105,6 +106,11 @@ def run(self) -> dict: xformers_version = xformers.__version__ + if get_python_version() >= (3, 10): + platform_info = f"{platform.freedesktop_os_release().get('PRETTY_NAME', None)} - {platform.platform()}" + else: + platform_info = platform.platform() + is_notebook_str = "Yes" if is_notebook() else "No" is_google_colab_str = "Yes" if is_google_colab() else "No" @@ -152,7 +158,7 @@ def run(self) -> dict: info = { "🤗 Diffusers version": version, - "Platform": platform.platform(), + "Platform": platform_info, "Running on a notebook?": is_notebook_str, "Running on Google Colab?": is_google_colab_str, "Python version": platform.python_version(),