fix: support ctranslate2 rev7 vocab files (.json) (#327)

2023-08-02 21:36:31 +08:00 · 2023-08-02 21:36:31 +08:00 · 6a50902ca7
parent b990781c9f
commit 6a50902ca7
1 changed files with 2 additions and 0 deletions
--- a/crates/tabby-download/src/lib.rs
+++ b/crates/tabby-download/src/lib.rs
@ -70,6 +70,8 @@ pub async fn download_model(model_id: &str, prefer_local_file: bool) -> Result<(
    let optional_files = vec![
        "ctranslate2/vocabulary.txt",
        "ctranslate2/shared_vocabulary.txt",
+        "ctranslate2/vocabulary.json",
+        "ctranslate2/shared_vocabulary.json",
    ];
    for path in optional_files {
        cache_info