From 74c896acfd591e8cfb80735159ee831dff70c7dc Mon Sep 17 00:00:00 2001
From: YdrMaster <ydrml@hotmail.com>
Date: Sun, 28 Apr 2024 09:27:22 +0800
Subject: [PATCH] docs: update README.md and xtask help

Signed-off-by: YdrMaster <ydrml@hotmail.com>
---
 Cargo.lock        | 12 ++++++------
 README.md         |  9 +++++++--
 xtask/src/main.rs |  2 +-
 3 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 92b2dc2c..e9d431fd 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2110,18 +2110,18 @@ checksum = "a3f0bf26fd526d2a95683cd0f87bf103b8539e2ca1ef48ce002d67aad59aa0b4"
 
 [[package]]
 name = "serde"
-version = "1.0.198"
+version = "1.0.199"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9846a40c979031340571da2545a4e5b7c4163bdae79b301d5f86d03979451fcc"
+checksum = "0c9f6e76df036c77cd94996771fb40db98187f096dd0b9af39c6c6e452ba966a"
 dependencies = [
  "serde_derive",
 ]
 
 [[package]]
 name = "serde_derive"
-version = "1.0.198"
+version = "1.0.199"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e88edab869b01783ba905e7d0153f9fc1a6505a96e4ad3018011eedb838566d9"
+checksum = "11bd257a6541e141e42ca6d24ae26f7714887b47e89aa739099104c7e4d3b7fc"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -2907,9 +2907,9 @@ checksum = "bec47e5bfd1bff0eeaf6d8b485cc1074891a197ab4225d504cb7a1ab88b02bf0"
 
 [[package]]
 name = "winnow"
-version = "0.6.6"
+version = "0.6.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f0c976aaaa0e1f90dbb21e9587cdaf1d9679a1cde8875c0d6bd83ab96a208352"
+checksum = "14b9415ee827af173ebb3f15f9083df5a122eb93572ec28741fb153356ea2578"
 dependencies = [
  "memchr",
 ]
diff --git a/README.md b/README.md
index de9ba635..3b393f8f 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,12 @@
 # 手写 transformer 模型
 
+![GitHub Issues or Pull Requests](https://img.shields.io/github/issues/InfiniTensor/transformer-rs)
+![GitHub Issues or Pull Requests](https://img.shields.io/github/issues-pr/InfiniTensor/transformer-rs)
+![GitHub repo size](https://img.shields.io/github/repo-size/InfiniTensor/transformer-rs)
+![GitHub code size in bytes](https://img.shields.io/github/languages/code-size/InfiniTensor/transformer-rs)
+![GitHub contributors](https://img.shields.io/github/contributors/InfiniTensor/transformer-rs)
+![GitHub commit activity](https://img.shields.io/github/commit-activity/m/InfiniTensor/transformer-rs)
+
 从 [YdrMaster/llama2.rs](https://github.com/YdrMaster/llama2.rs) 发展来的手写 transformer 模型项目。
 
 ## 使用
@@ -57,5 +64,3 @@ cargo generate --model <model> --prompt <prompt>
 - `prompt`: 生成文本的开头；
 
 其他参数参见 `cargo generate --help`。
-
-
diff --git a/xtask/src/main.rs b/xtask/src/main.rs
index d5f37c90..e7a64c49 100644
--- a/xtask/src/main.rs
+++ b/xtask/src/main.rs
@@ -66,7 +66,7 @@ struct InferenceArgs {
     top_p: Option<f32>,
 
     #[cfg(feature = "nvidia")]
-    /// Use Nvidia GPU.
+    /// Use Nvidia GPU, specify device IDs separated by comma, e.g. `0` or `0,1`.
     #[clap(long)]
     nvidia: Option<String>,
 }