diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-03-09 11:21:48 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-09 11:21:48 +0100 |
| commit | 56c9d3ee7b92bd1605be9f94416fa89976ebba87 (patch) | |
| tree | 2b833f52db05a3dc718b10802946f5aa6a7dff63 | |
| parent | dd00482ea3456111482ec1cee045d2ae8efaf8ba (diff) | |
| download | candle-56c9d3ee7b92bd1605be9f94416fa89976ebba87.tar.gz candle-56c9d3ee7b92bd1605be9f94416fa89976ebba87.tar.bz2 candle-56c9d3ee7b92bd1605be9f94416fa89976ebba87.zip | |
Fix the model path for rwkv. (#1825)
| -rw-r--r-- | candle-examples/examples/rwkv/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/rwkv/main.rs b/candle-examples/examples/rwkv/main.rs index e971a1cc..8fb2c0d4 100644 --- a/candle-examples/examples/rwkv/main.rs +++ b/candle-examples/examples/rwkv/main.rs @@ -141,7 +141,7 @@ impl std::fmt::Display for Which { impl Which { fn model_id(&self) -> &'static str { match self { - Self::Eagle7b => "RWKV/HF_v5-Eagle-7B", + Self::Eagle7b => "RWKV/v5-Eagle-7B-HF", Self::World1b5 => "RWKV/rwkv-5-world-1b5", Self::World3b => "RWKV/rwkv-5-world-3b", Self::World6_1b6 => "paperfun/rwkv", |