Skip to content

Commit 2fd7935

Browse files
chekaaaAUTOMATIC1111
authored andcommitted
Remove wrong self reference in CUDA support for invokeai
1 parent da72bec commit 2fd7935

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/sd_hijack_optimizations.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def einsum_op_cuda(q, k, v):
181181
mem_free_torch = mem_reserved - mem_active
182182
mem_free_total = mem_free_cuda + mem_free_torch
183183
# Divide factor of safety as there's copying and fragmentation
184-
return self.einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20))
184+
return einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20))
185185

186186
def einsum_op(q, k, v):
187187
if q.device.type == 'cuda':

0 commit comments

Comments
 (0)