SnowFlash383935 commited on
Commit
2af1c2d
·
verified ·
1 Parent(s): 034daed

Update model_config.py

Browse files
Files changed (1) hide show
  1. model_config.py +1 -1
model_config.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import PretrainedConfig
2
 
3
  class TabularTransformerConfig(PretrainedConfig):
4
- model_type = "fleshka_tabular_transformer"
5
 
6
  def __init__(
7
  self,
 
1
  from transformers import PretrainedConfig
2
 
3
  class TabularTransformerConfig(PretrainedConfig):
4
+ model_type = "transformer"
5
 
6
  def __init__(
7
  self,