mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-05 22:21:31 +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 = optimized_attention_for_device(query_states.device, mask=False)
|
||||||
|
|
||||||
attn_output = attn(
|
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)
|
attn_output = attn_output.transpose(1, 2)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user