Upload folder using huggingface_hub
Browse files- eat_model.py +1 -1
- modeling_eat.py +2 -2
eat_model.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch.nn as nn
|
|
| 3 |
from timm.models.layers import trunc_normal_
|
| 4 |
from functools import partial
|
| 5 |
import numpy as np
|
| 6 |
-
from model_core import (
|
| 7 |
PatchEmbed_new,
|
| 8 |
get_2d_sincos_pos_embed_flexible,
|
| 9 |
FixedPositionalEncoder,
|
|
|
|
| 3 |
from timm.models.layers import trunc_normal_
|
| 4 |
from functools import partial
|
| 5 |
import numpy as np
|
| 6 |
+
from .model_core import (
|
| 7 |
PatchEmbed_new,
|
| 8 |
get_2d_sincos_pos_embed_flexible,
|
| 9 |
FixedPositionalEncoder,
|
modeling_eat.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
# modeling_eat.py
|
| 2 |
|
| 3 |
from transformers import PreTrainedModel
|
| 4 |
-
from configuration_eat import EATConfig
|
| 5 |
-
from eat_model import EAT
|
| 6 |
|
| 7 |
class EATModel(PreTrainedModel):
|
| 8 |
config_class = EATConfig
|
|
|
|
| 1 |
# modeling_eat.py
|
| 2 |
|
| 3 |
from transformers import PreTrainedModel
|
| 4 |
+
from .configuration_eat import EATConfig
|
| 5 |
+
from .eat_model import EAT
|
| 6 |
|
| 7 |
class EATModel(PreTrainedModel):
|
| 8 |
config_class = EATConfig
|