From 9c7e6d5bbaa55205d0678369588c019108fb30a7 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 18 Feb 2023 11:31:02 -0500
Subject: [PATCH] store and print real torch version

---
 modules/ui.py | 7 ++++++-
 webui.py      | 1 +
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/modules/ui.py b/modules/ui.py
index f5df1ffe..d9df3781 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1779,10 +1779,15 @@ def versions_html():
     else:
         xformers_version = "N/A"
 
+    try:
+        torch_version = torch.__long_version__
+    except:
+        torch_version = torch.__version__
+
     return f"""
 python: <span title="{sys.version}">{python_version}</span>
  • 
-torch: {torch.__version__}
+torch: {torch_version}
  • 
 xformers: {xformers_version}
  • 
diff --git a/webui.py b/webui.py
index 5b5c2139..2363ea4e 100644
--- a/webui.py
+++ b/webui.py
@@ -20,6 +20,7 @@ import torch
 
 # Truncate version number of nightly/local build of PyTorch to not cause exceptions with CodeFormer or Safetensors
 if ".dev" in torch.__version__ or "+git" in torch.__version__:
+    torch.__long_version__ = torch.__version__
     torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
 
 from modules import shared, devices, sd_samplers, upscaler, extensions, localization, ui_tempdir, ui_extra_networks