Skip to content

Commit

Permalink
added bert-large-uncased model (#439)
Browse files Browse the repository at this point in the history
Co-authored-by: guillaume-be <[email protected]>
  • Loading branch information
kavan-mevada and guillaume-be authored Aug 18, 2024
1 parent eabebc6 commit 62b40d0
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions src/models/bert/bert_model.rs
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,11 @@ impl BertModelResources {
"bert/model",
"https://huggingface.co/bert-base-uncased/resolve/main/rust_model.ot",
);
/// Shared under Apache 2.0 license by the Google team at <https://github.com/google-research/bert>. Modified with conversion to C-array format.
pub const BERT_LARGE: (&'static str, &'static str) = (
"bert-large/model",
"https://huggingface.co/bert-large-uncased/resolve/main/rust_model.ot",
);
/// Shared under MIT license by the MDZ Digital Library team at the Bavarian State Library at <https://github.com/dbmdz/berts>. Modified with conversion to C-array format.
pub const BERT_NER: (&'static str, &'static str) = (
"bert-ner/model",
Expand Down Expand Up @@ -75,6 +80,11 @@ impl BertConfigResources {
"bert/config",
"https://huggingface.co/bert-base-uncased/resolve/main/config.json",
);
/// Shared under Apache 2.0 license by the Google team at <https://github.com/google-research/bert>. Modified with conversion to C-array format.
pub const BERT_LARGE: (&'static str, &'static str) = (
"bert-large/config",
"https://huggingface.co/bert-large-uncased/resolve/main/config.json",
);
/// Shared under MIT license by the MDZ Digital Library team at the Bavarian State Library at <https://github.com/dbmdz/berts>. Modified with conversion to C-array format.
pub const BERT_NER: (&'static str, &'static str) = (
"bert-ner/config",
Expand Down Expand Up @@ -108,6 +118,11 @@ impl BertVocabResources {
"bert/vocab",
"https://huggingface.co/bert-base-uncased/resolve/main/vocab.txt",
);
/// Shared under Apache 2.0 license by the Google team at <https://github.com/google-research/bert>. Modified with conversion to C-array format.
pub const BERT_LARGE: (&'static str, &'static str) = (
"bert-large/vocab",
"https://huggingface.co/bert-large-uncased/resolve/main/vocab.txt",
);
/// Shared under MIT license by the MDZ Digital Library team at the Bavarian State Library at <https://github.com/dbmdz/berts>. Modified with conversion to C-array format.
pub const BERT_NER: (&'static str, &'static str) = (
"bert-ner/vocab",
Expand Down

0 comments on commit 62b40d0

Please sign in to comment.