allow absoulte model and tokenizer paths

microsoft · May 10, 2024 · ae7fd27 · ae7fd27
1 parent 0064baa
commit ae7fd27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/controllers/aici_native/src/bintokens.rs b/controllers/aici_native/src/bintokens.rs
@@ -166,7 +166,7 @@ pub fn find_tokenizer(mut name: &str) -> Result<ByteTokenizer> {
 
     log::info!("loading tokenizer: {}", name);
 
-    let loaded = if name.starts_with(".") {
+    let loaded = if name.starts_with(".") || name.starts_with("/") {
         Tokenizer::from_file(name)
     } else {
         let mut name2 = name.to_string();

diff --git a/rllm/rllm-base/src/server/mod.rs b/rllm/rllm-base/src/server/mod.rs
@@ -579,7 +579,7 @@ pub async fn server_main<ME: ModelExec>(
         args.revision = Some(rev);
     }
 
-    if args.model.starts_with(".") {
+    if args.model.starts_with(".") || args.model.starts_with("/") {
         args.local_weights = Some(args.model.clone());
     }