From 45c46f4cb3d6924882bd944712be168c7c2f605d Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Wed, 21 Sep 2022 10:32:58 +0300 Subject: [PATCH] oops --- launch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/launch.py b/launch.py index 72d2a5dc..b34dc714 100644 --- a/launch.py +++ b/launch.py @@ -106,7 +106,7 @@ if not is_installed("torch"): run(f'"{python}" -m {torch_command}', "Installing torch", "Couldn't install torch") if not skip_torch_cuda_test: - run_python("import torch; assert not torch.cuda.is_available(), 'Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDINE_ARGS variable to disable this check'") + run_python("import torch; assert torch.cuda.is_available(), 'Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDINE_ARGS variable to disable this check'") if not is_installed("k_diffusion.sampling"): run_pip(f"install {k_diffusion_package}", "k-diffusion")