extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-23 08:43:42 +01:00
Code Issues Packages Projects Releases Wiki Activity

Moved up to not duplicate if statement.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-28 02:05:48 -04:00
parent 0185ef7c83
commit 02240bda25
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/sd_internal/runtime.py
View File

@ -74,11 +74,11 @@ def device_select(device):
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) 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: if thread_data.force_full_precision:
print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', device_name) print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', device_name)
# Apply force_full_precision now.
thread_data.precision = 'full'
thread_data.device = device thread_data.device = device
thread_data.has_valid_gpu = True thread_data.has_valid_gpu = True
# Apply force_full_precision if was set.
thread_data.precision = 'full' if thread_data.force_full_precision else 'autocast'
return True return True
def device_init(device_selection=None): def device_init(device_selection=None):