Commit ·
1f55a56
1
Parent(s): 1e1c006
fix
Browse files- __pycache__/attn.cpython-312.pyc +0 -0
- attn.py +1 -1
__pycache__/attn.cpython-312.pyc
CHANGED
|
Binary files a/__pycache__/attn.cpython-312.pyc and b/__pycache__/attn.cpython-312.pyc differ
|
|
|
attn.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import math
|
| 2 |
-
from torch.nn.attention.flex_attention import flex_attention, create_block_mask
|
| 3 |
import torch
|
|
|
|
| 4 |
import torch.nn as nn
|
| 5 |
import torch.nn.functional as F
|
| 6 |
|
|
|
|
| 1 |
import math
|
|
|
|
| 2 |
import torch
|
| 3 |
+
from torch.nn.attention.flex_attention import flex_attention, create_block_mask
|
| 4 |
import torch.nn as nn
|
| 5 |
import torch.nn.functional as F
|
| 6 |
|