readme fix

tracel-ai · Feb 9, 2024 · fdbd45a · fdbd45a
1 parent 9e4f98a
commit fdbd45a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/bert-burn/README.md b/bert-burn/README.md
@@ -1,8 +1,7 @@
 # Bert-Burn Model
 
 This project provides an example implementation for inference on the BERT family of models. The following compatible
-bert-variants: `roberta-base`**(default)
-**/`roberta-large`, `bert-base-uncased`/`bert-large-uncased`/`bert-base-cased`/`bert-large-cased`
+bert-variants: `roberta-base`(**default**)/`roberta-large`, `bert-base-uncased`/`bert-large-uncased`/`bert-base-cased`/`bert-large-cased`
 can be loaded as following. The pre-trained weights and config files are automatically downloaded
 from: [HuggingFace Model hub](https://huggingface.co/FacebookAI/roberta-base/tree/main)
 

diff --git a/bert-burn/examples/infer-embedding.rs b/bert-burn/examples/infer-embedding.rs
@@ -18,7 +18,7 @@ pub fn launch<B: Backend>(device: B::Device) {
     let default_model = "roberta-base".to_string();
     let model_variant = if args.len() > 1 {
         // Use the argument provided by the user
-        // Possible values: "roberta-base", "roberta-large"
+        // Possible values: "bert-base-uncased", "roberta-large" etc.
         &args[1]
     } else {
         // Use the default value if no argument is provided