viljami
/
sdw
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove wrong self reference in CUDA support for invokeai

Browse Source
master
Cheka 3 years ago committed by AUTOMATIC1111
parent da72becb13
commit 2fd7935ef4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
modules/sd_hijack_optimizations.py
Unescape Escape View File

@ -181,7 +181,7 @@ def einsum_op_cuda(q, k, v):
mem_free_torch = mem_reserved - mem_active mem_free_torch = mem_reserved - mem_active
mem_free_total = mem_free_cuda + mem_free_torch mem_free_total = mem_free_cuda + mem_free_torch
# Divide factor of safety as there's copying and fragmentation # Divide factor of safety as there's copying and fragmentation
return self.einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20)) return einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20))
def einsum_op(q, k, v): def einsum_op(q, k, v):
if q.device.type == 'cuda': if q.device.type == 'cuda':

Write Preview
Loading…
Cancel
Save