From 37b293fe745dd5c7a48daffd35470cd79c733793 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Wed, 28 Dec 2022 17:46:24 +0530 Subject: [PATCH] Force full precision on NVIDIA T400 --- ui/easydiffusion/device_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/easydiffusion/device_manager.py b/ui/easydiffusion/device_manager.py index 252aeee4..b7406408 100644 --- a/ui/easydiffusion/device_manager.py +++ b/ui/easydiffusion/device_manager.py @@ -125,7 +125,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) and (' 1660' in device_name or ' 1650' in device_name)) or ('Quadro T2000' in device_name) + return (('nvidia' in device_name or 'geforce' in device_name) and (' 1660' in device_name or ' 1650' in device_name or ' t400' in device_name)) or ('Quadro T2000' in device_name) def get_max_vram_usage_level(device): if device != 'cpu':