holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Replaced missing gpu_name by device_name

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-22 21:28:12 -04:00
parent 28fed6281f
commit 0d62123a0b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/runtime.py
View File

@ -73,7 +73,7 @@ def device_select(device):
# otherwise these NVIDIA cards create green images
thread_data.force_full_precision = ('nvidia' in device_name.lower() or 'geforce' in device_name.lower()) and (' 1660' in device_name or ' 1650' in device_name)
if thread_data.force_full_precision:
print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', gpu_name)
print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', device_name)
thread_data.device = device
thread_data.has_valid_gpu = True