Upload model
Browse files- modeling.py +1 -1
modeling.py
CHANGED
|
@@ -18,7 +18,7 @@ from transformers.utils.backbone_utils import load_backbone
|
|
| 18 |
from .configuration import LSPDetrConfig, STAConfig
|
| 19 |
|
| 20 |
|
| 21 |
-
flex_attention = torch.compile(flex_attention, dynamic=
|
| 22 |
|
| 23 |
|
| 24 |
def init_freqs(head_dim: int, num_heads: int, pos_dim: int, theta: float) -> Tensor:
|
|
|
|
| 18 |
from .configuration import LSPDetrConfig, STAConfig
|
| 19 |
|
| 20 |
|
| 21 |
+
flex_attention = torch.compile(flex_attention, dynamic=True)
|
| 22 |
|
| 23 |
|
| 24 |
def init_freqs(head_dim: int, num_heads: int, pos_dim: int, theta: float) -> Tensor:
|