Commit aa7ddb8b authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub

Merge pull request #7911 from vladmandic/torch-version

store and print real torch version
parents 3715ece0 9c7e6d5b
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment