Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Seunghoon Lee
2023-04-26 12:34:27 +09:00
parent 09ae33cdf7
commit 8b75033a11
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_hijack_optimizations.py
+1 -1
View File
@@ -202,7 +202,7 @@ def einsum_op_cuda(q, k, v):
return einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20))
def einsum_op_dml(q, k, v):
mem_total, mem_active = devices.adl.memory_stats()
mem_total, mem_active = torch.dml.memory_stats(q.device)
mem_reserved = mem_total / (1 << 20) * 0.7
return einsum_op_tensor_mem(q, k, v, (mem_reserved - mem_active) if mem_reserved > mem_active else 1)