ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add resume args in webui

Browse Source 
Former-commit-id: 1d86ad768b1f36e54b4c2a9f18f6ea5a7df04c90
This commit is contained in:
hiyouga
2024-06-08 00:22:16 +08:00
parent f45e81e186
commit 3f6b3eed98
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: