Unverified Commit aa7ddb8b authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #7911 from vladmandic/torch-version 

store and print real torch version
parents 3715ece0 9c7e6d5b

modules/ui.py

+6 −1
Original line number Diff line number Diff line
@@ -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}

 •

webui.py

+1 −0
Original line number Diff line number Diff line
@@ -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

CRA Git | Maintained and supported by SUSTech CRA and CCSE