mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-04 13:40:52 +08:00
Dinov3: disable sage
This commit is contained in:
parent
9e4794da5c
commit
24a9bb5b79
@ -77,7 +77,8 @@ class DINOv3ViTAttention(nn.Module):
|
||||
attn = optimized_attention_for_device(query_states.device, mask=False)
|
||||
|
||||
attn_output = attn(
|
||||
query_states, key_states, value_states, self.num_heads, attention_mask, skip_reshape=True, skip_output_reshape=True
|
||||
query_states, key_states, value_states, self.num_heads, attention_mask,
|
||||
skip_reshape=True, skip_output_reshape=True, low_precision_attention=False,
|
||||
)
|
||||
|
||||
attn_output = attn_output.transpose(1, 2)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user