423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 15:52:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

modify style

Browse Source 
Former-commit-id: c1f1df99e4dc3d0aadf1207b4e9a16218187fd5a
This commit is contained in:
BUAADreamer 2024-04-25 21:29:50 +08:00
parent 110c2ce2a5
commit 058ed5e607
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
from peft import LoraConfig, LoraModel, PeftModel, TaskType, get_peft_model
@ -21,11 +21,11 @@ logger = get_logger(__name__)
def init_adapter(
config: "PretrainedConfig",
model: Union["PreTrainedModel"],
model: "PreTrainedModel",
model_args: "ModelArguments",
finetuning_args: "FinetuningArguments",
is_trainable: bool,
) -> Union["PreTrainedModel"]:
) -> "PreTrainedModel":
r"""
Initializes the adapters.

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

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