42 lines
918 B
Rust
42 lines
918 B
Rust
|
use std::env;
|
||
|
|
use std::path::PathBuf;
|
||
|
|
|
||
|
|
fn get_root_dir() -> PathBuf {
|
||
|
|
match env::var("TABBY_ROOT") {
|
||
|
|
Ok(x) => PathBuf::from(x),
|
||
|
|
Err(_) => PathBuf::from(env::var("HOME").unwrap()).join(".tabby"),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct ModelDir(PathBuf);
|
||
|
|
|
||
|
|
impl ModelDir {
|
||
|
|
pub fn new(model: &str) -> Self {
|
||
|
|
Self(get_root_dir().join("models").join(model))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn from(path: &str) -> Self {
|
||
|
|
Self(PathBuf::from(path))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn path(&self) -> &PathBuf {
|
||
|
|
&self.0
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn path_string(&self, name: &str) -> String {
|
||
|
|
self.0.join(name).display().to_string()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn metadata_file(&self) -> String {
|
||
|
|
return self.path_string("metadata.json");
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn tokenizer_file(&self) -> String {
|
||
|
|
return self.path_string("tokenizer.json");
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn ctranslate2_dir(&self) -> String {
|
||
|
|
self.path_string("ctranslate2")
|
||
|
|
}
|
||
|
|
}
|