AXCXEPT commited on
Commit
44cfc2a
1 Parent(s): 312533f

Update modeling_custom_qwen.py

Browse files
Files changed (1) hide show
  1. modeling_custom_qwen.py +1 -1
modeling_custom_qwen.py CHANGED
@@ -10,7 +10,7 @@ import re
10
  logging.basicConfig(level=logging.INFO)
11
 
12
  class CustomQwen2Config(Qwen2Config):
13
- model_type = "custom_qwen2config"
14
 
15
  def __init__(self, **kwargs):
16
  super().__init__(**kwargs)
 
10
  logging.basicConfig(level=logging.INFO)
11
 
12
  class CustomQwen2Config(Qwen2Config):
13
+ model_type = "custom_qwen2"
14
 
15
  def __init__(self, **kwargs):
16
  super().__init__(**kwargs)