Spaces:
Runtime error
Runtime error
rm unused import
Browse files
models.py
CHANGED
|
@@ -15,7 +15,7 @@ import numpy as np
|
|
| 15 |
import torch
|
| 16 |
import torch.nn as nn
|
| 17 |
import torch.nn.functional as F
|
| 18 |
-
from torch.nn.attention.flex_attention import BlockMask, flex_attention
|
| 19 |
|
| 20 |
|
| 21 |
class Mlp(nn.Module):
|
|
|
|
| 15 |
import torch
|
| 16 |
import torch.nn as nn
|
| 17 |
import torch.nn.functional as F
|
| 18 |
+
# from torch.nn.attention.flex_attention import BlockMask, flex_attention
|
| 19 |
|
| 20 |
|
| 21 |
class Mlp(nn.Module):
|