add yaml to manifest

This commit is contained in:
Hardik Shah 2024-07-22 13:34:08 -07:00
parent 6f0d348b1c
commit 74442e88b1
3 changed files with 7 additions and 3 deletions

View file

@ -0,0 +1,9 @@
inference_config:
impl_type: "inline"
inline_config:
checkpoint_type: "pytorch"
checkpoint_dir: {checkpoint_dir}/
tokenizer_path: {checkpoint_dir}/tokenizer.model
model_parallel_size: {model_parallel_size}
max_seq_len: 2048
max_batch_size: 1