zhiqu22
commited on
Commit
·
9be44fc
1
Parent(s):
5805e00
update codes
Browse files- modeling_mitre.py +1 -1
modeling_mitre.py
CHANGED
|
@@ -5,7 +5,7 @@ from typing import List, Optional, Tuple, Union, Dict, Any
|
|
| 5 |
|
| 6 |
import torch
|
| 7 |
from torch import nn
|
| 8 |
-
from configuration_mitre import MitreConfig
|
| 9 |
from transformers.utils import logging
|
| 10 |
|
| 11 |
from transformers.generation import GenerationMixin
|
|
|
|
| 5 |
|
| 6 |
import torch
|
| 7 |
from torch import nn
|
| 8 |
+
from .configuration_mitre import MitreConfig
|
| 9 |
from transformers.utils import logging
|
| 10 |
|
| 11 |
from transformers.generation import GenerationMixin
|