Match the config class name to what the modeling code expects
Browse files
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
|