Skip to content

Commit 12a8d7f

Browse files
blessedcoolantlstein
authored andcommitted
Fix crash introduced in #1866
1 parent 3d2b497 commit 12a8d7f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ldm/models/diffusion/cross_attention_control.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ def einsum_op_cuda(self, q, k, v):
280280
return self.einsum_op_slice_dim1(q, k, v, slice_size)
281281

282282
# fallback for when there is no saved strategy, or saved strategy does not slice
283-
mem_free_total = self.cached_mem_free_total or get_mem_free_total(q.device)
283+
mem_free_total = get_mem_free_total(q.device)
284284
# Divide factor of safety as there's copying and fragmentation
285285
return self.einsum_op_tensor_mem(q, k, v, mem_free_total / 3.3 / (1 << 20))
286286

0 commit comments

Comments
 (0)