FlameF0X commited on
Commit
7fe5689
·
verified ·
1 Parent(s): 860a775

Update modeling_i3.py

Browse files
Files changed (1) hide show
  1. modeling_i3.py +1 -1
modeling_i3.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from torch import nn
3
  from transformers import PreTrainedModel, PretrainedConfig
4
- from i3_modules import i3Model # import your original i3Model class
5
 
6
  class i3Config(PretrainedConfig):
7
  model_type = "i3"
 
1
  import torch
2
  from torch import nn
3
  from transformers import PreTrainedModel, PretrainedConfig
4
+ from i3_modules import i3Model # import the original i3Model class
5
 
6
  class i3Config(PretrainedConfig):
7
  model_type = "i3"