deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-04 09:49:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

update docs of llamafactory (#825)

Browse Source
This commit is contained in:
Zhangchi Feng
2025-02-11 10:11:42 +08:00
committed by GitHub
parent 47283856a3
commit d364570bff
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docs/llamafactory_train_and_infer.md
View File

@@ -29,7 +29,7 @@ mkdir configs # let's put all yaml files here
## Dataset Prepare ## Dataset Prepare
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. 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).
### Image Dataset ### Image Dataset
@@ -265,7 +265,7 @@ lora_target: q_proj,v_proj
### dataset ### dataset
dataset: mllm_demo # mllm_demo mllm_video_demo mllm_audio_demo dataset: mllm_demo # mllm_demo mllm_video_demo mllm_audio_demo
template: minicpm_v template: minicpm_o # minicpm_o minicpm_v
cutoff_len: 3072 cutoff_len: 3072
max_samples: 1000 max_samples: 1000
overwrite_cache: true overwrite_cache: true
@@ -313,7 +313,7 @@ llamafactory-cli export configs/minicpmo_2_6_lora_export.yaml
### model ### model
model_name_or_path: openbmb/MiniCPM-o-2_6 # MiniCPM-o-2_6 MiniCPM-V-2_6 model_name_or_path: openbmb/MiniCPM-o-2_6 # MiniCPM-o-2_6 MiniCPM-V-2_6
adapter_name_or_path: saves/minicpmo_2_6/lora/sft adapter_name_or_path: saves/minicpmo_2_6/lora/sft
template: minicpm_v template: minicpm_o # minicpm_o minicpm_v
finetuning_type: lora finetuning_type: lora
trust_remote_code: true trust_remote_code: true
@@ -355,7 +355,7 @@ deepspeed: configs/deepspeed/ds_z2_config.json
### dataset ### dataset
dataset: mllm_demo # mllm_demo mllm_video_demo dataset: mllm_demo # mllm_demo mllm_video_demo
template: minicpm_v template: minicpm_o # minicpm_o minicpm_v
cutoff_len: 3072 cutoff_len: 3072
max_samples: 1000 max_samples: 1000
overwrite_cache: true overwrite_cache: true
@@ -404,7 +404,7 @@ CUDA_VISIBLE_DEVICES=0 llamafactory-cli webchat configs/minicpmo_2_6_infer.yaml
```yaml ```yaml
model_name_or_path: saves/minicpmo_2_6/full/sft model_name_or_path: saves/minicpmo_2_6/full/sft
template: minicpm_v template: minicpm_o # minicpm_o minicpm_v
infer_backend: huggingface infer_backend: huggingface
trust_remote_code: true trust_remote_code: true
``` ```