Skip to content

Commit d364570

Browse files
authored
update docs of llamafactory (OpenBMB#825)
1 parent 4728385 commit d364570

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

docs/llamafactory_train_and_infer.md

+5-5
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ mkdir configs # let's put all yaml files here
2929

3030
## Dataset Prepare
3131

32-
Refer to [data/dataset_info.json](https://github.com/hiyouga/LLaMA-Factory/blob/main/data/dataset_info.json) to add your customised dataset. Let's use the two existing demo datasets `mllm_demo` and `mllm_video_demo` as examples.
32+
Refer to [data/dataset_info.json](https://github.com/hiyouga/LLaMA-Factory/blob/main/data/dataset_info.json) to add your customised dataset. Let's use the two existing demo datasets `mllm_demo`, `mllm_video_demo` and `mllm_audio_demo` as examples (audio is only for MiniCPM-o-2.6).
3333

3434
### Image Dataset
3535

@@ -265,7 +265,7 @@ lora_target: q_proj,v_proj
265265

266266
### dataset
267267
dataset: mllm_demo # mllm_demo mllm_video_demo mllm_audio_demo
268-
template: minicpm_v
268+
template: minicpm_o # minicpm_o minicpm_v
269269
cutoff_len: 3072
270270
max_samples: 1000
271271
overwrite_cache: true
@@ -313,7 +313,7 @@ llamafactory-cli export configs/minicpmo_2_6_lora_export.yaml
313313
### model
314314
model_name_or_path: openbmb/MiniCPM-o-2_6 # MiniCPM-o-2_6 MiniCPM-V-2_6
315315
adapter_name_or_path: saves/minicpmo_2_6/lora/sft
316-
template: minicpm_v
316+
template: minicpm_o # minicpm_o minicpm_v
317317
finetuning_type: lora
318318
trust_remote_code: true
319319

@@ -355,7 +355,7 @@ deepspeed: configs/deepspeed/ds_z2_config.json
355355

356356
### dataset
357357
dataset: mllm_demo # mllm_demo mllm_video_demo
358-
template: minicpm_v
358+
template: minicpm_o # minicpm_o minicpm_v
359359
cutoff_len: 3072
360360
max_samples: 1000
361361
overwrite_cache: true
@@ -404,7 +404,7 @@ CUDA_VISIBLE_DEVICES=0 llamafactory-cli webchat configs/minicpmo_2_6_infer.yaml
404404

405405
```yaml
406406
model_name_or_path: saves/minicpmo_2_6/full/sft
407-
template: minicpm_v
407+
template: minicpm_o # minicpm_o minicpm_v
408408
infer_backend: huggingface
409409
trust_remote_code: true
410410
```

0 commit comments

Comments
 (0)