From 35f752b36db135ee87b393618824ffeb24ab93f4 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Fri, 31 Jan 2025 16:28:12 +0530 Subject: [PATCH] Move the half precision bug check logic to sdkit --- ui/easydiffusion/device_manager.py | 17 +++-------------- 1 file changed, 3 insertions(+), 14 deletions(-) diff --git a/ui/easydiffusion/device_manager.py b/ui/easydiffusion/device_manager.py index 9b0f3f5e..8196a883 100644 --- a/ui/easydiffusion/device_manager.py +++ b/ui/easydiffusion/device_manager.py @@ -6,6 +6,8 @@ import traceback import torch from easydiffusion.utils import log +from sdkit.utils import has_half_precision_bug + """ Set `FORCE_FULL_PRECISION` in the environment variables, or in `config.bat`/`config.sh` to set full precision (i.e. float32). Otherwise the models will load at half-precision (i.e. float16). @@ -160,20 +162,7 @@ def needs_to_force_full_precision(context): return True device_name = context.device_name.lower() - return ( - ("nvidia" in device_name or "geforce" in device_name or "quadro" in device_name) - and ( - " 1660" in device_name - or " 1650" in device_name - or " 1630" in device_name - or " t400" in device_name - or " t550" in device_name - or " t600" in device_name - or " t1000" in device_name - or " t1200" in device_name - or " t2000" in device_name - ) - ) or ("tesla k40m" in device_name) + return has_half_precision_bug(device_name) def get_max_vram_usage_level(device):