Skip to content

Commit f216273

Browse files
修改了部分书写bug
1. 修改了cookbook中的虚拟环境地址 2. 修改重复依赖
1 parent 33953b1 commit f216273

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

PROMPT.md

+1
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@ data[:5]
132132
<|assistant|>
133133
该文件看起来包含有关某些条目的元数据,每个条目有以下字段:
134134
135+
135136
- `file_name`: 文件名称
136137
- `name`: 名称
137138
- `type`: 类型(例如 "survivor" 或 "killer")

cookbook/finetune_muti_classfication.ipynb

+2-2
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@
178178
"from transformers import AutoModel, AutoTokenizer\n",
179179
"import torch\n",
180180
"# 参数设置\n",
181-
"model_path = \"/Models/chatglm3-6b-base\"\n",
181+
"model_path = \"THUDM/chatglm3-6b-base\"\n",
182182
"tokenizer_path = model_path\n",
183183
"device = \"cuda\"\n",
184184
"\n",
@@ -1973,7 +1973,7 @@
19731973
"os.makedirs(output_dir, exist_ok=True)\n",
19741974
"# 构建命令\n",
19751975
"command = f\"\"\"\n",
1976-
"/home/zr/Code/ChatGLM3/venv/bin/torchrun --standalone --nnodes=1 --nproc_per_node={num_gpus} ../finetune_basemodel_demo/finetune.py \\\n",
1976+
"torchrun --standalone --nnodes=1 --nproc_per_node={num_gpus} ../finetune_basemodel_demo/finetune.py \\\n",
19771977
" --train_format input-output \\\n",
19781978
" --train_file {dataset_path} \\\n",
19791979
" --lora_rank {lora_rank} \\\n",

requirements.txt

-1
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,4 @@ sse-starlette
99
streamlit>=1.24.0
1010
fastapi>=0.104.1
1111
uvicorn~=0.24.0
12-
sse_starlette
1312
loguru~=0.7.2

0 commit comments

Comments
 (0)