FlameF0X commited on
Commit
2c1f948
·
verified ·
1 Parent(s): 33e68fe

Update configuration_i3.py

Browse files
Files changed (1) hide show
  1. configuration_i3.py +2 -1
configuration_i3.py CHANGED
@@ -1,7 +1,8 @@
 
1
  from transformers import PretrainedConfig
2
 
3
  class I3Config(PretrainedConfig):
4
- model_type = "i3" # this tells Transformers your model type is "i3"
5
 
6
  def __init__(
7
  self,
 
1
+ # configuration_i3.py
2
  from transformers import PretrainedConfig
3
 
4
  class I3Config(PretrainedConfig):
5
+ model_type = "i3" # unique name for your model
6
 
7
  def __init__(
8
  self,