Skip to content

Commit 37ecf4c

Browse files
committed
[ppdiffusers] fix uvit recompute
1 parent ecc5557 commit 37ecf4c

File tree

1 file changed

+1
-1
lines changed
  • ppdiffusers/ppdiffusers/models

1 file changed

+1
-1
lines changed

ppdiffusers/ppdiffusers/models/uvit.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def __init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0.
5454
self.proj = nn.Linear(dim, dim)
5555
self.proj_drop = nn.Dropout(proj_drop)
5656

57-
self._use_memory_efficient_attention_xformers = True
57+
self._use_memory_efficient_attention_xformers = is_ppxformers_available()
5858
self._attention_op = None
5959

6060
def reshape_heads_to_batch_dim(self, tensor, transpose=True):

0 commit comments

Comments
 (0)