Add missing key to init_kwargs

Former-commit-id: 03fc4621dad132164596a58d3e8693787b7e1aca
This commit is contained in:
Yaser Afshar
2024-10-25 10:15:42 -07:00
committed by hiyouga
parent 8881237475
commit 6f1c8dacea
11 changed files with 15 additions and 10 deletions

View File

@@ -152,6 +152,7 @@ class Runner:
bf16=(get("train.compute_type") == "bf16"),
pure_bf16=(get("train.compute_type") == "pure_bf16"),
plot_loss=True,
trust_remote_code=True,
ddp_timeout=180000000,
include_num_input_tokens_seen=False if is_transformers_version_equal_to_4_46() else True, # FIXME
)
@@ -268,6 +269,7 @@ class Runner:
top_p=get("eval.top_p"),
temperature=get("eval.temperature"),
output_dir=get_save_dir(model_name, finetuning_type, get("eval.output_dir")),
trust_remote_code=True,
)
if get("eval.predict"):