mohsennp commited on
Commit
68f220e
·
verified ·
1 Parent(s): 8e2efae

Upload DeCodon

Browse files
Files changed (2) hide show
  1. config.json +1 -0
  2. configuration_decodon.py +2 -0
config.json CHANGED
@@ -21,6 +21,7 @@
21
  "layer_norm_eps": 1e-12,
22
  "lm_type": "distilled",
23
  "max_position_embeddings": 2048,
 
24
  "num_attention_heads": 16,
25
  "num_hidden_layers": 12,
26
  "pad_token_id": 3,
 
21
  "layer_norm_eps": 1e-12,
22
  "lm_type": "distilled",
23
  "max_position_embeddings": 2048,
24
+ "model_type": "decodon",
25
  "num_attention_heads": 16,
26
  "num_hidden_layers": 12,
27
  "pad_token_id": 3,
configuration_decodon.py CHANGED
@@ -1,6 +1,8 @@
1
  from transformers import PretrainedConfig
2
 
3
  class DeCodonConfig(PretrainedConfig):
 
 
4
  def __init__(
5
  self,
6
  vocab_size=70,
 
1
  from transformers import PretrainedConfig
2
 
3
  class DeCodonConfig(PretrainedConfig):
4
+ model_type = "decodon"
5
+
6
  def __init__(
7
  self,
8
  vocab_size=70,