Log device usage stats during thread startup

This commit is contained in:
cmdr2 2023-05-02 18:26:29 +05:30
parent 729f7eb24a
commit eaba64a64a

View File

@ -17,6 +17,7 @@ from sdkit.utils import (
img_to_base64_str,
img_to_buffer,
latent_samples_to_images,
get_device_usage,
)
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"
)
log.info("Device usage during initialization:")
get_device_usage(device, log_info=True)
device_manager.device_init(context, device)