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

prevent flooding the log with warnings for GPU<3GB

Browse Source
This commit is contained in:
JeLuF 2023-01-04 02:45:51 +01:00
parent 2d9853f1f4
commit 60f7c73c8a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ui/easydiffusion/device_manager.py
View File

@ -156,6 +156,8 @@ def is_device_compatible(device):
''' '''
Returns True/False, and prints any compatibility errors Returns True/False, and prints any compatibility errors
''' '''
# static variable "history".
is_device_compatible.history = getattr(is_device_compatible, 'history', {})
try: try:
validate_device_id(device, log_prefix='is_device_compatible') validate_device_id(device, log_prefix='is_device_compatible')
except: except:
@ -168,7 +170,9 @@ def is_device_compatible(device):
_, mem_total = torch.cuda.mem_get_info(device) _, mem_total = torch.cuda.mem_get_info(device)
mem_total /= float(10**9) mem_total /= float(10**9)
if mem_total < 3.0: if mem_total < 3.0:
log.warn(f'GPU {device} with less than 3 GB of VRAM is not compatible with Stable Diffusion') if is_device_compatible.history.get(device) == None:
log.warn(f'GPU {device} with less than 3 GB of VRAM is not compatible with Stable Diffusion')
is_device_compatible.history[device] = 1
return False return False
except RuntimeError as e: except RuntimeError as e:
log.error(str(e)) log.error(str(e))