From fec21f1208cea51a90b8be6ea83647a1032b1a2b Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sun, 11 Sep 2022 11:10:54 +0530 Subject: [PATCH] Catch 16xx cards without the NVIDIA name in them --- ui/sd_internal/runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index afcacb34..92c5e889 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -60,7 +60,7 @@ try: gpu_name = torch.cuda.get_device_name(torch.cuda.current_device()) print('detected: ', gpu_name) has_valid_gpu = True - force_full_precision = ('nvidia' in gpu_name.lower()) and ('1660' in gpu_name or ' 1650' in gpu_name) # otherwise these NVIDIA cards create green images + force_full_precision = ('nvidia' in gpu_name.lower() or 'geforce' in gpu_name.lower()) and (' 1660' in gpu_name or ' 1650' in gpu_name) # otherwise these NVIDIA cards create green images if force_full_precision: print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', gpu_name) except: