Update modeling_custom_qwen.py
Browse files- 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 = "
|
| 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)
|