huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the model path for rwkv. (#1825)

Browse Source
This commit is contained in:
Laurent Mazare
2024-03-09 11:21:48 +01:00
committed by GitHub
parent dd00482ea3
commit 56c9d3ee7b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/rwkv/main.rs
View File

@ -141,7 +141,7 @@ impl std::fmt::Display for Which {
impl Which { impl Which {
fn model_id(&self) -> &'static str { fn model_id(&self) -> &'static str {
match self { match self {
Self::Eagle7b => "RWKV/HF_v5-Eagle-7B", Self::Eagle7b => "RWKV/v5-Eagle-7B-HF",
Self::World1b5 => "RWKV/rwkv-5-world-1b5", Self::World1b5 => "RWKV/rwkv-5-world-1b5",
Self::World3b => "RWKV/rwkv-5-world-3b", Self::World3b => "RWKV/rwkv-5-world-3b",
Self::World6_1b6 => "paperfun/rwkv", Self::World6_1b6 => "paperfun/rwkv",