From fcdc3f2dd0402cca7c8c52f529e523f8c23cb179 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 11 Jun 2024 18:27:21 +0530 Subject: [PATCH] Don't crash if psutils fails to get cpu or memory usage --- scripts/check_modules.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index 85c6036c..755841bf 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -128,10 +128,10 @@ def update_modules(): # if sdkit is 2.0.15.x (or lower), then diffusers should be restricted to 0.21.4 (see below for the reason) # otherwise use the current sdkit version (with the corresponding diffusers version) - expected_sdkit_version_str = "2.0.20.3" + expected_sdkit_version_str = "2.0.20.4" expected_diffusers_version_str = "0.28.2" - legacy_sdkit_version_str = "2.0.15.6" + legacy_sdkit_version_str = "2.0.15.7" legacy_diffusers_version_str = "0.21.4" sdkit_version_str = version("sdkit") @@ -413,9 +413,17 @@ def launch_uvicorn(): setup_amd_environment() print("\nLaunching uvicorn\n") - + import uvicorn - uvicorn.run("main:server_api", port=listen_port, log_level="error", app_dir=os.environ["SD_UI_PATH"], host=bind_ip, access_log=False) + + uvicorn.run( + "main:server_api", + port=listen_port, + log_level="error", + app_dir=os.environ["SD_UI_PATH"], + host=bind_ip, + access_log=False, + ) ### Start