diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index 9e98634f..0aeb1993 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -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