diff options
| -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", |