We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ecc5557 commit 37ecf4cCopy full SHA for 37ecf4c
ppdiffusers/ppdiffusers/models/uvit.py
@@ -54,7 +54,7 @@ def __init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0.
54
self.proj = nn.Linear(dim, dim)
55
self.proj_drop = nn.Dropout(proj_drop)
56
57
- self._use_memory_efficient_attention_xformers = True
+ self._use_memory_efficient_attention_xformers = is_ppxformers_available()
58
self._attention_op = None
59
60
def reshape_heads_to_batch_dim(self, tensor, transpose=True):
0 commit comments