diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-08-08 11:39:34 +0900 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-08 11:39:34 +0900 |
| commit | f17c8c2eff63210f5e96e1e2b049b46ba9cfa389 (patch) | |
| tree | 701056aec9ae11bc45df9b39b176a54fa4d34e19 /modules/errors.py | |
| parent | c75bda867be5345bf959daf23bdc19eadc90841a (diff) | |
| parent | 01997f45ba089af24b03a5f614147bb0f9d8d824 (diff) | |
Merge branch 'dev' into auro-autolaunch
Diffstat (limited to 'modules/errors.py')
| -rw-r--r-- | modules/errors.py | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/modules/errors.py b/modules/errors.py index dffabe45..192cd8ff 100644 --- a/modules/errors.py +++ b/modules/errors.py @@ -84,3 +84,53 @@ def run(code, task): code()
except Exception as e:
display(task, e)
+
+
+def check_versions():
+ from packaging import version
+ from modules import shared
+
+ import torch
+ import gradio
+
+ expected_torch_version = "2.0.0"
+ expected_xformers_version = "0.0.20"
+ expected_gradio_version = "3.39.0"
+
+ if version.parse(torch.__version__) < version.parse(expected_torch_version):
+ print_error_explanation(f"""
+You are running torch {torch.__version__}.
+The program is tested to work with torch {expected_torch_version}.
+To reinstall the desired version, run with commandline flag --reinstall-torch.
+Beware that this will cause a lot of large files to be downloaded, as well as
+there are reports of issues with training tab on the latest version.
+
+Use --skip-version-check commandline argument to disable this check.
+ """.strip())
+
+ if shared.xformers_available:
+ import xformers
+
+ if version.parse(xformers.__version__) < version.parse(expected_xformers_version):
+ print_error_explanation(f"""
+You are running xformers {xformers.__version__}.
+The program is tested to work with xformers {expected_xformers_version}.
+To reinstall the desired version, run with commandline flag --reinstall-xformers.
+
+Use --skip-version-check commandline argument to disable this check.
+ """.strip())
+
+ if gradio.__version__ != expected_gradio_version:
+ print_error_explanation(f"""
+You are running gradio {gradio.__version__}.
+The program is designed to work with gradio {expected_gradio_version}.
+Using a different version of gradio is extremely likely to break the program.
+
+Reasons why you have the mismatched gradio version can be:
+ - you use --skip-install flag.
+ - you use webui.py to start the program instead of launch.py.
+ - an extension installs the incompatible gradio version.
+
+Use --skip-version-check commandline argument to disable this check.
+ """.strip())
+
|