Update configuration_plm.py
Browse files- configuration_plm.py +4 -4
configuration_plm.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# coding=utf-8
|
2 |
-
# Copyright 2024 The
|
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 |
-
"""
|
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
|
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 = "
|
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__(
|