From 67cca3bc007a2a2fa8171e42711be278a4cfe7c4 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Mon, 7 Nov 2022 18:26:10 +0530 Subject: [PATCH] Print the devices for which rendering threads have started; Prettier print of the model data --- ui/sd_internal/runtime.py | 7 ++++++- ui/server.py | 2 ++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index 6fba4ddd..c003d356 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -285,7 +285,12 @@ def load_model_ckpt(): thread_data.model_is_half = False thread_data.model_fs_is_half = False - print('loaded', thread_data.ckpt_file, 'as', model.device, '->', modelCS.cond_stage_model.device, '->', thread_data.modelFS.device, 'using precision', thread_data.precision) + print(f'''loaded model + model file: {thread_data.ckpt_file}.ckpt + model.device: {model.device} + modelCS.device: {modelCS.cond_stage_model.device} + modelFS.device: {thread_data.modelFS.device} + using precision: {thread_data.precision}''') def unload_filters(): if thread_data.model_gfpgan is not None: diff --git a/ui/server.py b/ui/server.py index 50098cc1..bf5871a4 100644 --- a/ui/server.py +++ b/ui/server.py @@ -440,5 +440,7 @@ if is_using_a_gpu and task_manager.is_alive(0) <= 0: print('Add the line "@set CUDA_VISIBLE_DEVICES=N" where N is the GPUs to use to config.bat') print('Add the line "CUDA_VISIBLE_DEVICES=N" where N is the GPUs to use to config.sh') +print('active devices', task_manager.get_devices()) + # start the browser ui import webbrowser; webbrowser.open('http://localhost:9000')