Print GPU name

2022-09-11 10:59:45 +05:30 · 2022-09-11 10:59:45 +05:30 · 1d4b34c0dd
commit 1d4b34c0dd
parent 3668c87e0d
1 changed files with 1 additions and 0 deletions
--- a/ui/sd_internal/runtime.py
+++ b/ui/sd_internal/runtime.py
@ -58,6 +58,7 @@ has_valid_gpu = False
 force_full_precision = False
 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
    if force_full_precision: