Text Generation
PyTorch
English
Chinese
plm
conversational
custom_code
UCASLuoyang commited on
Commit
06c8d07
·
verified ·
1 Parent(s): 45c134e

Update configuration_plm.py

Browse files
Files changed (1) hide show
  1. configuration_plm.py +4 -4
configuration_plm.py CHANGED
@@ -1,5 +1,5 @@
1
  # coding=utf-8
2
- # Copyright 2024 The EdgeLLM team and The HuggingFace Inc. All rights reserved.
3
  #
4
  # Licensed under the Apache License, Version 2.0 (the "License");
5
  # you may not use this file except in compliance with the License.
@@ -12,7 +12,7 @@
12
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
  # See the License for the specific language governing permissions and
14
  # limitations under the License.
15
- """EdgeLLM model configuration"""
16
  # Test test
17
  from transformers.configuration_utils import PretrainedConfig
18
  from transformers.utils import logging
@@ -21,7 +21,7 @@ from transformers.utils import logging
21
  logger = logging.get_logger(__name__)
22
 
23
 
24
- class EdgellmConfig(PretrainedConfig):
25
  r"""
26
  This is the configuration class to store the configuration of a [`Qwen2Model`]. It is used to instantiate a
27
  Qwen2 model according to the specified arguments, defining the model architecture. Instantiating a configuration
@@ -88,7 +88,7 @@ class EdgellmConfig(PretrainedConfig):
88
  >>> configuration = model.config
89
  ```"""
90
 
91
- model_type = "edgellm"
92
  keys_to_ignore_at_inference = ["past_key_values"]
93
 
94
  def __init__(
 
1
  # coding=utf-8
2
+ # Copyright 2024 The PLM team and The HuggingFace Inc. All rights reserved.
3
  #
4
  # Licensed under the Apache License, Version 2.0 (the "License");
5
  # you may not use this file except in compliance with the License.
 
12
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
  # See the License for the specific language governing permissions and
14
  # limitations under the License.
15
+ """PLM model configuration"""
16
  # Test test
17
  from transformers.configuration_utils import PretrainedConfig
18
  from transformers.utils import logging
 
21
  logger = logging.get_logger(__name__)
22
 
23
 
24
+ class PLMConfig(PretrainedConfig):
25
  r"""
26
  This is the configuration class to store the configuration of a [`Qwen2Model`]. It is used to instantiate a
27
  Qwen2 model according to the specified arguments, defining the model architecture. Instantiating a configuration
 
88
  >>> configuration = model.config
89
  ```"""
90
 
91
+ model_type = "plm"
92
  keys_to_ignore_at_inference = ["past_key_values"]
93
 
94
  def __init__(