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

add resume args in webui

Browse Source 
Former-commit-id: 06e5d136a4
This commit is contained in:
hiyouga
2024-06-08 00:22:16 +08:00
parent d79222894c
commit a8318723a4
7 changed files with 68 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/model/adapter.py
View File

@@ -50,7 +50,7 @@ def init_adapter(
logger.info("Upcasting trainable params to float32.")
cast_trainable_params_to_fp32 = True
if finetuning_args.finetuning_type == "full" and is_trainable:
if is_trainable and finetuning_args.finetuning_type == "full":
logger.info("Fine-tuning method: Full")
forbidden_modules = set()
@@ -67,7 +67,7 @@ def init_adapter(
else:
param.requires_grad_(False)
if finetuning_args.finetuning_type == "freeze" and is_trainable:
if is_trainable and finetuning_args.finetuning_type == "freeze":
logger.info("Fine-tuning method: Freeze")
if model_args.visual_inputs: