File tree Expand file tree Collapse file tree
examples/singa_peft/src/singa_peft Expand file tree Collapse file tree 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