WordPiece should use vocab.txt and not vocab.json
@@ -252,8 +252,8 @@ impl Model for WordPiece {
fn save(&self, folder: &Path, name: Option<&str>) -> Result<Vec<PathBuf>> {
let vocab_file_name = match name {
- Some(name) => format!("{}-vocab.json", name),
- None => "vocab.json".to_string(),
+ Some(name) => format!("{}-vocab.txt", name),
+ None => "vocab.txt".to_string(),
};
// Write vocab.txt
