extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-23 13:51:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #411 from madrang/beta

Browse Source 
Apply force_full_precision if was set on device_select.
This commit is contained in:
cmdr2 2022-10-28 11:50:16 +05:30 committed by GitHub
commit 2432491bfc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/runtime.py
View File

@ -74,6 +74,8 @@ 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)
if thread_data.force_full_precision:
print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', device_name)
# Apply force_full_precision now before models are loaded.
thread_data.precision = 'full'
thread_data.device = device
thread_data.has_valid_gpu = True