Dinov3: disable sage

This commit is contained in:
kijai 2026-05-22 19:48:46 +03:00
parent 9e4794da5c
commit 24a9bb5b79

View File

@ -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)