From 09e069ae6ca700cabb885662a83a05227b4f79a5 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 20 May 2024 06:22:29 -0400
Subject: [PATCH] Log the pytorch version.

---
 comfy/model_management.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 21ae8d29..b2192862 100644
--- 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: