lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-07 00:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve hints

Browse Source
This commit is contained in:
layerdiffusion
2024-08-11 02:00:53 -07:00
parent 33187c26f1
commit 86ee2d9462
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/launch_utils.py
View File

@@ -432,8 +432,10 @@ def prepare_environment():
args.skip_torch_cuda_test = True
if not args.skip_torch_cuda_test and not check_run_python("import torch; assert torch.cuda.is_available()"):
raise RuntimeError(
'Torch is not able to use GPU; '
'add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check'
'Your device does not support the current version of Torch/CUDA! Consider download another version: \n'
'https://github.com/lllyasviel/stable-diffusion-webui-forge/releases/tag/latest'
# 'Torch is not able to use GPU; '
# 'add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check'
)
startup_timer.record("torch GPU test")