extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 09:57:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Log device usage stats during thread startup

Browse Source
This commit is contained in:
cmdr2 2023-05-02 18:26:29 +05:30
parent 729f7eb24a
commit eaba64a64a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/easydiffusion/renderer.py
View File

@ -17,6 +17,7 @@ from sdkit.utils import (
img_to_base64_str, img_to_base64_str,
img_to_buffer, img_to_buffer,
latent_samples_to_images, latent_samples_to_images,
get_device_usage,
) )
context = Context() # thread-local context = Context() # thread-local
@ -40,6 +41,9 @@ def init(device):
app_config.get("test_diffusers", False) and app_config.get("update_branch", "main") != "main" app_config.get("test_diffusers", False) and app_config.get("update_branch", "main") != "main"
) )
log.info("Device usage during initialization:")
get_device_usage(device, log_info=True)
device_manager.device_init(context, device) device_manager.device_init(context, device)