This commit is contained in:
Disty0
2023-04-30 18:37:58 +03:00
parent 56cdac6592
commit 185b796991
-1
View File
@@ -44,7 +44,6 @@ def get_memory():
mem.update({ 'ram': e })
try:
import torch
from modules import shared
if torch.cuda.is_available():
s = torch.cuda.mem_get_info()
gpu = { 'free': gb(s[0]), 'used': gb(s[1] - s[0]), 'total': gb(s[1]) }