store and print real torch version
This commit is contained in:
parent
3715ece0ad
commit
9c7e6d5bba
2 changed files with 7 additions and 1 deletions
|
@ -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}
|
||||
•
|
||||
|
|
1
webui.py
1
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
|
||||
|
|
Loading…
Reference in a new issue