diff options
author | MalumaDev <piano.lu92@gmail.com> | 2022-10-19 10:22:39 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-19 10:22:39 +0200 |
commit | 2362d5f00eeecbe6a6f67fe6085da01a3d78e407 (patch) | |
tree | 763d0a836b63cc0155c4ca810f12db3af57eb448 /modules/sd_hijack_optimizations.py | |
parent | c2765c9bcd264f5a8922348cd03521cb2ff306b3 (diff) | |
parent | 1b91cbbc1163d3613aa329bed3aecd8e29ca52ca (diff) |
Merge branch 'master' into test_resolve_conflicts
Diffstat (limited to 'modules/sd_hijack_optimizations.py')
-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':
|