File tree Expand file tree Collapse file tree 1 file changed +33
-0
lines changed
examples/singa_peft/src/singa_peft Expand file tree Collapse file tree 1 file changed +33
-0
lines changed Original file line number Diff line number Diff line change
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
You can’t perform that action at this time.
0 commit comments