JustinDuc commited on
Commit
f31997e
·
verified ·
1 Parent(s): 9c93c06

Update saute_model.py

Browse files
Files changed (1) hide show
  1. saute_model.py +1 -1
saute_model.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  import torch.nn as nn
3
  from transformers import PreTrainedModel, BertModel, BertTokenizerFast
4
  from transformers.modeling_outputs import MaskedLMOutput
5
- from sources.saute_config import SAUTEConfig
6
 
7
  activation_to_class = {
8
  "gelu" : nn.GELU,
 
2
  import torch.nn as nn
3
  from transformers import PreTrainedModel, BertModel, BertTokenizerFast
4
  from transformers.modeling_outputs import MaskedLMOutput
5
+ from .saute_config import SAUTEConfig
6
 
7
  activation_to_class = {
8
  "gelu" : nn.GELU,