worstchan commited on
Commit
7fafb17
·
verified ·
1 Parent(s): 25d88d3

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. eat_model.py +1 -1
  2. 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