From d6890a4501b9a2eb8d7e9353cded80e6dad96985 Mon Sep 17 00:00:00 2001
From: YdrMaster <ydrml@hotmail.com>
Date: Tue, 27 Feb 2024 09:32:56 +0800
Subject: [PATCH] docs: update README.md

Signed-off-by: YdrMaster <ydrml@hotmail.com>
---
 README.md             | 7 ++++---
 xtask/src/generate.rs | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index a0277538..b3503cf1 100644
--- a/README.md
+++ b/README.md
@@ -9,10 +9,10 @@
 ### 启动文本生成
 
 ```plaintext
-cargo generate --model <model> --prompt <prompt> [--log <log>]
+cargo generate --model <model> --prompt <prompt>
 ```
 
-参数：
+必要参数：
 
 - `model`: 存放模型文件的目录，至少包含以下 3 个文件：
   - `config.json`: 模型配置文件；
@@ -20,7 +20,8 @@ cargo generate --model <model> --prompt <prompt> [--log <log>]
   - `tokenizer.model`: 分词器词表；
     > 目前仅支持 32000 词 BPE Tonkenizer。
 - `prompt`: 生成文本的起始文本。
-- `log` (optional, default = "warn"): 日志级别，可为 `off`/`trace`/`debug`/`info`/`warn`/`error`。
+
+其他参数参见 `cargo generate --help`。
 
 ### 转换参数
 
diff --git a/xtask/src/generate.rs b/xtask/src/generate.rs
index 00ca5cf8..eb21a042 100644
--- a/xtask/src/generate.rs
+++ b/xtask/src/generate.rs
@@ -25,7 +25,7 @@ pub(crate) struct GenerateArgs {
     /// Copy model parameters inside memory.
     #[clap(short, long)]
     inside_mem: bool,
-    /// Log level.
+    /// Log level, may be "off", "trace", "debug", "info" or "error".
     #[clap(short, long)]
     log: Option<String>,
 }