Log the pytorch version.

2024-05-20 06:22:29 -04:00 · 2024-05-20 06:22:29 -04:00 · 09e069ae6c
parent 11a2ad5110
commit 09e069ae6c
1 changed files with 5 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -120,6 +120,11 @@ total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
 total_ram = psutil.virtual_memory().total / (1024 * 1024)
 logging.info("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram))

+try:
+    logging.info("pytorch version: {}".format(torch.version.__version__))
+except:
+    pass
+
 try:
    OOM_EXCEPTION = torch.cuda.OutOfMemoryError
 except: