From 8c6fa0075c4d15456f117c32a5fe4c045b07292d Mon Sep 17 00:00:00 2001 From: Charles Gao <64410968+SchweitzerGAO@users.noreply.github.com> Date: Sat, 15 Apr 2023 10:14:32 +0800 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9a2c02f..95d80b1 100644 --- a/README.md +++ b/README.md @@ -67,9 +67,9 @@ python3.6、 transformers==4.2.0、pytorch==1.7.0 ### Quick Start 在[模型分享](#model_share)中下载模型,将模型文件夹model_epoch40_50w放到model目录下,执行如下命令,进行对话 ``` -python interact.py --no_cuda --model_path model_epoch40_50w (使用cpu生成,速度相对较慢) +python interact.py --no_cuda --model_path ./model/model_epoch40_50w (使用cpu生成,速度相对较慢) 或 -python interact.py --model_path model_epoch40_50w --device 0 (指定0号GPU进行生成,速度相对较快) +python interact.py --model_path ./model/model_epoch40_50w --device 0 (指定0号GPU进行生成,速度相对较快) ```