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

Print GPU name

Browse Source
This commit is contained in:
cmdr2 2022-09-11 10:59:45 +05:30
parent 3668c87e0d
commit 1d4b34c0dd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ui/sd_internal/runtime.py
View File

@ -58,6 +58,7 @@ has_valid_gpu = False
force_full_precision = False force_full_precision = False
try: try:
gpu_name = torch.cuda.get_device_name(torch.cuda.current_device()) gpu_name = torch.cuda.get_device_name(torch.cuda.current_device())
print('detected: ', gpu_name)
has_valid_gpu = True has_valid_gpu = True
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 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: if force_full_precision: