mirror of
https://github.com/hiyouga/LlamaFactory.git
synced 2026-03-19 23:33:09 +00:00
[model] update constants (#10220)
This commit is contained in:
@@ -110,6 +110,7 @@ def _freeze_model_parameters(model: Any, finetuning_args: "FinetuningArguments")
|
||||
if any(name.startswith(k) for k in params_to_freeze):
|
||||
p.requires_grad_(False)
|
||||
|
||||
|
||||
def run_pt(
|
||||
model_args: "ModelArguments",
|
||||
data_args: "DataArguments",
|
||||
|
||||
Reference in New Issue
Block a user