Spaces:
Build error
Build error
Update lxmert/src/lxmert_lrp.py
Browse files- lxmert/src/lxmert_lrp.py +1 -1
lxmert/src/lxmert_lrp.py
CHANGED
|
@@ -24,7 +24,7 @@ from typing import Optional, Tuple
|
|
| 24 |
import torch
|
| 25 |
from torch import nn
|
| 26 |
from torch.nn import CrossEntropyLoss, SmoothL1Loss
|
| 27 |
-
from
|
| 28 |
from transformers.file_utils import (
|
| 29 |
ModelOutput,
|
| 30 |
add_code_sample_docstrings,
|
|
|
|
| 24 |
import torch
|
| 25 |
from torch import nn
|
| 26 |
from torch.nn import CrossEntropyLoss, SmoothL1Loss
|
| 27 |
+
from .layers import *
|
| 28 |
from transformers.file_utils import (
|
| 29 |
ModelOutput,
|
| 30 |
add_code_sample_docstrings,
|