saracandu commited on
Commit
f5e5ec4
·
verified ·
1 Parent(s): e2fe7e1
Files changed (1) hide show
  1. modeling_stlenc.py +1 -1
modeling_stlenc.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  import torch.nn as nn
3
  from transformers import PreTrainedModel
4
- from configuration_stlenc import STLEncoderConfig
5
 
6
  class STLEncoderModel(PreTrainedModel):
7
  config_class = STLEncoderConfig
 
1
  import torch
2
  import torch.nn as nn
3
  from transformers import PreTrainedModel
4
+ from .configuration_stlenc import STLEncoderConfig
5
 
6
  class STLEncoderModel(PreTrainedModel):
7
  config_class = STLEncoderConfig