From 11e1436e2efb92b41aae84289addc4ab3b1b9171 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 19 May 2023 17:56:20 +0530
Subject: [PATCH] 2 GB cards aren't exactly 2 GB
---
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 3a820116..dc705927 100644
--- a/ui/easydiffusion/device_manager.py
+++ b/ui/easydiffusion/device_manager.py
@@ -225,7 +225,7 @@ def is_device_compatible(device):
try:
_, mem_total = torch.cuda.mem_get_info(device)
mem_total /= float(10**9)
- if mem_total < 2.0:
+ if mem_total < 1.9:
if is_device_compatible.history.get(device) == None:
log.warn(f"GPU {device} with less than 2 GB of VRAM is not compatible with Stable Diffusion")
is_device_compatible.history[device] = 1