make tokeziner name required for dev ex (#718)

predibase · Dec 16, 2024 · 170aae1 · 170aae1
1 parent abf9f39
commit 170aae1
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,10 +1,5 @@
 [workspace]
-members = [
-    "router",
-    "router/client",
-    "router/grpc-metadata",
-    "launcher"
-]
+members = ["router", "router/client", "router/grpc-metadata", "launcher"]
 
 [workspace.package]
 version = "0.1.0"

diff --git a/router/Cargo.toml b/router/Cargo.toml
@@ -20,6 +20,7 @@ axum = { version = "0.7", features = ["json", "macros"] }
 axum-tracing-opentelemetry = "0.16"
 clap = { version = "4.1.4", features = ["derive", "env"] }
 futures = "0.3.26"
+home = "=0.5.9"
 hf-hub = { version = "0.3.0", features = ["tokio"] }
 h2 = "0.3.26"
 lorax-client = { path = "client" }

diff --git a/router/src/main.rs b/router/src/main.rs
@@ -58,7 +58,7 @@ struct Args {
     port: u16,
     #[clap(default_value = "/tmp/lorax-server-0", long, env)]
     master_shard_uds_path: String,
-    #[clap(default_value = "bigscience/bloom", long, env)]
+    #[clap(long, env)]
     tokenizer_name: String,
     #[clap(long, env)]
     tokenizer_config_path: Option<String>,