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

Use CPU mode for graphics cards with less than 3 GB of RAM

Browse Source
This commit is contained in:
cmdr2 2022-09-11 10:34:04 +05:30
parent a1e5a2cb67
commit d88e0f16ac
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ui/sd_internal/runtime.py
View File

@ -57,11 +57,19 @@ sampler_ddim = None
has_valid_gpu = False
force_full_precision = False
try:
gpu_name = torch.cuda.get_device_name(torch.cuda.current_device())
has_valid_gpu = True
gpu = torch.cuda.current_device()
gpu_name = torch.cuda.get_device_name(gpu)
force_full_precision = ('nvidia' in gpu_name.lower()) and ('1660' in gpu_name or ' 1650' in gpu_name) # otherwise these NVIDIA cards create green images
if force_full_precision:
print('forcing full precision on NVIDIA 16xx cards, to avoid green images. GPU detected: ', gpu_name)
mem_free, mem_total = torch.cuda.mem_get_info(gpu)
mem_total /= float(10**9)
if mem_total < 3.0:
print("GPUs with less than 3 GB of VRAM are not compatible with Stable Diffusion")
raise Exception()
has_valid_gpu = True
except:
print('WARNING: No compatible GPU found. Using the CPU, but this will be very slow!')
pass