From 2c359e0e399acff8b475666cd47c255edd7aec97 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 27 Jun 2025 15:20:30 +0530
Subject: [PATCH] webui: logging changes

---
 ui/easydiffusion/backends/webui/__init__.py | 4 +++-
 ui/easydiffusion/backends/webui/impl.py     | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ui/easydiffusion/backends/webui/__init__.py b/ui/easydiffusion/backends/webui/__init__.py
index 68eb66ba..fc491bf2 100644
--- a/ui/easydiffusion/backends/webui/__init__.py
+++ b/ui/easydiffusion/backends/webui/__init__.py
@@ -349,7 +349,9 @@ def get_env():
         "PIP_INSTALLER_LOCATION": [],  # [f"{dir}/python/get-pip.py"],
         "TRANSFORMERS_CACHE": [f"{dir}/transformers-cache"],
         "HF_HUB_DISABLE_SYMLINKS_WARNING": ["true"],
-        "COMMANDLINE_ARGS": [f'--api --models-dir "{models_dir}" {model_path_args} --skip-torch-cuda-test'],
+        "COMMANDLINE_ARGS": [
+            f'--api --models-dir "{models_dir}" {model_path_args} --skip-torch-cuda-test --disable-gpu-warning'
+        ],
         "SKIP_VENV": ["1"],
         "SD_WEBUI_RESTARTING": ["1"],
     }
diff --git a/ui/easydiffusion/backends/webui/impl.py b/ui/easydiffusion/backends/webui/impl.py
index a651a66f..afc5cd9e 100644
--- a/ui/easydiffusion/backends/webui/impl.py
+++ b/ui/easydiffusion/backends/webui/impl.py
@@ -253,7 +253,7 @@ def generate_images(
         res = res.json()
     else:
         raise Exception(
-            "The engine failed while generating this image. Please check the logs in the command-line window for more details."
+            f"HTTP Status {res.status_code}. The engine failed while generating this image. Please check the logs in the command-line window for more details."
         )
 
     import json