mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2025-06-19 17:39:16 +02:00
Log device usage stats during thread startup
This commit is contained in:
parent
729f7eb24a
commit
eaba64a64a
@ -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)
|
||||
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user