diff options
| author | Cheka <carlos93@hotmail.cl> | 2022-10-18 20:28:28 -0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-19 09:35:53 +0300 |
| commit | 2fd7935ef4ed296db5dfd8c7fea99244816f8cf0 (patch) | |
| tree | 32d74e80b67fd6654774778c0478295e72875ace | |
| parent | da72becb13e4b750fbcb3d158c3f843311ef9938 (diff) | |
Remove wrong self reference in CUDA support for invokeai
| -rw-r--r-- | modules/sd_hijack_optimizations.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_hijack_optimizations.py b/modules/sd_hijack_optimizations.py index a3345bb9..98123fbf 100644 --- a/modules/sd_hijack_optimizations.py +++ b/modules/sd_hijack_optimizations.py @@ -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':
|