Fix dependencies.

Ai00-X · Dec 10, 2023 · a018d46 · a018d46
1 parent 783e1ca
commit a018d46
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -10,6 +10,7 @@ repository = "https://github.com/cgisky1980/ai00_rwkv_server"
 keywords = ["deep-learning", "LLM", "model", "rwkv"]
 categories = ["science"]
 exclude = ["assets"]
+default-run = "ai00_server"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
@@ -43,7 +44,10 @@ sha2 = "0.10.8"
 [dependencies.web-rwkv]
 git = "https://github.com/cryscan/web-rwkv"
 tag = "v0.4.5"
-features = ["converter"]
+
+[dependencies.web-rwkv-converter]
+git = "https://github.com/cryscan/web-rwkv-converter"
+tag = "v0.1.0"
 
 [dependencies.axum]
 version = "0.7.1"

diff --git a/src/bin/converter.rs b/src/bin/converter.rs
@@ -3,7 +3,7 @@ use std::{fs::File, path::PathBuf};
 use anyhow::Result;
 use clap::Parser;
 use memmap2::Mmap;
-use web_rwkv::converter::{convert_safetensors, RENAME, TRANSPOSE};
+use web_rwkv_converter::{convert_safetensors, RENAME, TRANSPOSE};
 
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)]