Skip to content

GitLab

Commit 8b0d81c6 authored by wanglch's avatar wanglch
Browse files

Update README.md

parent 5cb526ba
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
README.md
View file @ 8b0d81c6
...@@ -97,13 +97,13 @@ pip install -e . ...@@ -97,13 +97,13 @@ pip install -e .
--output_dir # 训练输出文件夹 --output_dir # 训练输出文件夹
``` ```
### 单卡训练 ### 单机单卡
``` ```
sh lora_finetune_single.sh sh lora_finetune_single.sh
``` ```
### 多卡训练 ### 单机多卡
``` ```
sh lora_finetune_multi.sh sh lora_finetune_multi.sh
...@@ -115,6 +115,8 @@ sh lora_finetune_multi.sh ...@@ -115,6 +115,8 @@ sh lora_finetune_multi.sh
### 单机单卡 ### 单机单卡
推理时需修改推理脚本中的模型地址为本地模型地址
`tokenizer = AutoTokenizer.from_pretrained("/home/wanglch/projects/GLM-4V/glm-4v-9b", trust_remote_code=True)` `tokenizer = AutoTokenizer.from_pretrained("/home/wanglch/projects/GLM-4V/glm-4v-9b", trust_remote_code=True)`
`model = AutoModelForCausalLM.from_pretrained( `model = AutoModelForCausalLM.from_pretrained(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment