Xuezha commited on
Commit
70ffc36
·
verified ·
1 Parent(s): 29586e8

Update configure.py

Browse files
Files changed (1) hide show
  1. configure.py +2 -2
configure.py CHANGED
@@ -2,8 +2,8 @@
2
 
3
  from transformers import PretrainedConfig
4
 
5
- class CustomTransformerConfig(PretrainedConfig):
6
- model_type = "custom_transformer"
7
 
8
  def __init__(self, embed_dim=1024, num_heads=8, num_layers=6, vocab_size=151646, **kwargs):
9
  super().__init__(**kwargs)
 
2
 
3
  from transformers import PretrainedConfig
4
 
5
+ class RecombinationTransformerConfig(PretrainedConfig):
6
+ model_type = "RecombinationTransformer"
7
 
8
  def __init__(self, embed_dim=1024, num_heads=8, num_layers=6, vocab_size=151646, **kwargs):
9
  super().__init__(**kwargs)