diff options
| -rw-r--r-- | modules/ui.py | 7 | ||||
| -rw-r--r-- | webui.py | 1 |
2 files changed, 7 insertions, 1 deletions
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}
•
@@ -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
|