Safetensors
llama

Match the config class name to what the modeling code expects

#4
Files changed (1) hide show
  1. configuration_deepseek.py +1 -1
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 DeepseekConfig(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
 
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