facebook
/

bart-base

Feature Extraction

Model card Files Files and versions

nielsr HF Staff commited on Sep 16, 2021

Commit

7652e0c

·

1 Parent(s): 51f9983

Fix code example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,8 +26,8 @@ Here is how to use this model in PyTorch:
 ```python
 from transformers import BartTokenizer, BartModel
-tokenizer = BartTokenizer.from_pretrained('facebook/bart-large')
-model = BartModel.from_pretrained('facebook/bart-large')
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 outputs = model(**inputs)

 ```python
 from transformers import BartTokenizer, BartModel
+tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
+model = BartModel.from_pretrained('facebook/bart-base')
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 outputs = model(**inputs)