Match the config class name to what the modeling code expects
#4
by
winglian
- opened
configuration_deepseek.py
CHANGED
@@ -4,7 +4,7 @@ from transformers.utils import logging
|
|
4 |
logger = logging.get_logger(__name__)
|
5 |
|
6 |
DEEPSEEK_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
|
7 |
-
class
|
8 |
r"""
|
9 |
This is the configuration class to store the configuration of a [`DeepseekModel`]. It is used to instantiate an DeepSeek
|
10 |
model according to the specified arguments, defining the model architecture. Instantiating a configuration with the
|
|
|
4 |
logger = logging.get_logger(__name__)
|
5 |
|
6 |
DEEPSEEK_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
|
7 |
+
class DeepseekV2Config(PretrainedConfig):
|
8 |
r"""
|
9 |
This is the configuration class to store the configuration of a [`DeepseekModel`]. It is used to instantiate an DeepSeek
|
10 |
model according to the specified arguments, defining the model architecture. Instantiating a configuration with the
|