2 files changed, 2 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py
index f5df1ffe..54efb6a4 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1782,7 +1782,7 @@ def versions_html():
     return f"""
 python: <span title="{sys.version}">{python_version}</span>
  • 
-torch: {torch.__version__}
+torch: {getattr(torch, '__long_version__',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