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

@@ -1,3 +1,4 @@
model_name_or_path: saves/llama3-8b/full/sft
template: llama3
infer_backend: huggingface # choices: [huggingface, vllm]
trust_remote_code: true