Skip to content

Commit

Permalink
Remove wrong self reference in CUDA support for invokeai
Browse files Browse the repository at this point in the history
  • Loading branch information
chekaaa authored and AUTOMATIC1111 committed Oct 19, 2022
1 parent da72bec commit 2fd7935
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/sd_hijack_optimizations.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,7 @@ def einsum_op_cuda(q, k, v):
mem_free_torch = mem_reserved - mem_active
mem_free_total = mem_free_cuda + mem_free_torch
# 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):
if q.device.type == 'cuda':
Expand Down

0 comments on commit 2fd7935

Please sign in to comment.