Spaces:
Running
Running
Commit
·
6e61c34
1
Parent(s):
ee1a8a3
Style: Fix formatting for models and tests
Browse files- src/models/encoder.py +1 -0
src/models/encoder.py
CHANGED
|
@@ -20,6 +20,7 @@ from typing import List, Optional, Tuple, Union
|
|
| 20 |
import torch
|
| 21 |
import torch.nn as nn
|
| 22 |
|
|
|
|
| 23 |
from .attention import MultiHeadAttention
|
| 24 |
from .feedforward import FeedForward
|
| 25 |
from .positional_encoding import PositionalEncoding
|
|
|
|
| 20 |
import torch
|
| 21 |
import torch.nn as nn
|
| 22 |
|
| 23 |
+
# Encoder implementation
|
| 24 |
from .attention import MultiHeadAttention
|
| 25 |
from .feedforward import FeedForward
|
| 26 |
from .positional_encoding import PositionalEncoding
|