Skip to content

Commit a3de028

Browse files
authored
Merge pull request apache#1303 from solopku/patch-16
Add the PEFT config file
2 parents 01362bb + 49a6966 commit a3de028

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
#
2+
# Licensed to the Apache Software Foundation (ASF) under one
3+
# or more contributor license agreements. See the NOTICE file
4+
# distributed with this work for additional information
5+
# regarding copyright ownership. The ASF licenses this file
6+
# to you under the Apache License, Version 2.0 (the
7+
# "License"); you may not use this file except in compliance
8+
# with the License. You may obtain a copy of the License at
9+
#
10+
# http://www.apache.org/licenses/LICENSE-2.0
11+
#
12+
# Unless required by applicable law or agreed to in writing,
13+
# software distributed under the License is distributed on an
14+
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15+
# KIND, either express or implied. See the License for the
16+
# specific language governing permissions and limitations
17+
# under the License.
18+
#
19+
20+
class PeftConfig:
21+
"""
22+
PeftConfig: the base class of all PEFT methods config, defines the common configuration parameters for all PEFT methods.
23+
"""
24+
def __init__(self, peft_type: str, freeze_base_model: bool = True):
25+
r"""
26+
peft_type: the type of peft, linear_lora etc...
27+
freeze_base_model: freeze base model parameters, default true
28+
Args:
29+
peft_type:
30+
freeze_base_model:
31+
"""
32+
self.peft_type = peft_type
33+
self.freeze_base_model = freeze_base_model

0 commit comments

Comments
 (0)