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')