michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #7911 from vladmandic/torch-version

Browse source 
store and print real torch version
This commit is contained in:
AUTOMATIC1111 2023-02-19 08:33:40 +03:00 committed by GitHub
commit aa7ddb8b0c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modules/ui.py
View file

@ -1779,10 +1779,15 @@ def versions_html():
else: else:
xformers_version = "N/A" xformers_version = "N/A"
try:
torch_version = torch.__long_version__
except:
torch_version = torch.__version__
return f""" return f"""
python: <span title="{sys.version}">{python_version}</span> python: <span title="{sys.version}">{python_version}</span>
 •   • 
torch: {torch.__version__} torch: {torch_version}
 •   • 
xformers: {xformers_version} xformers: {xformers_version}
 •   • 

1
webui.py
View file

@ -20,6 +20,7 @@ import torch
# Truncate version number of nightly/local build of PyTorch to not cause exceptions with CodeFormer or Safetensors # 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__: 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) 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 from modules import shared, devices, sd_samplers, upscaler, extensions, localization, ui_tempdir, ui_extra_networks