423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 13:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

modify style

Browse Source 
Former-commit-id: 235b4113709fe788b4f1a1a3089ce8356940877b
This commit is contained in:
BUAADreamer 2024-04-25 21:29:50 +08:00
parent 549f35b1fd
commit 40bfe767f7
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llmtuner/model/adapter.py
View File

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING, Union from typing import TYPE_CHECKING
import torch import torch
from peft import LoraConfig, LoraModel, PeftModel, TaskType, get_peft_model from peft import LoraConfig, LoraModel, PeftModel, TaskType, get_peft_model
@ -21,11 +21,11 @@ logger = get_logger(__name__)
def init_adapter( def init_adapter(
config: "PretrainedConfig", config: "PretrainedConfig",
model: Union["PreTrainedModel"], model: "PreTrainedModel",
model_args: "ModelArguments", model_args: "ModelArguments",
finetuning_args: "FinetuningArguments", finetuning_args: "FinetuningArguments",
is_trainable: bool, is_trainable: bool,
) -> Union["PreTrainedModel"]: ) -> "PreTrainedModel":
r""" r"""
Initializes the adapters. Initializes the adapters.

2
src/llmtuner/model/loader.py
View File

@ -112,7 +112,7 @@ def load_model(
finetuning_args: "FinetuningArguments", finetuning_args: "FinetuningArguments",
is_trainable: bool = False, is_trainable: bool = False,
add_valuehead: bool = False, add_valuehead: bool = False,
) -> Union["PreTrainedModel"]: ) -> "PreTrainedModel":
r""" r"""
Loads pretrained model. Loads pretrained model.
""" """