Fixing typos.
Browse files
README.md
CHANGED
|
@@ -83,10 +83,10 @@ from transformers import AutoModel, AutoTokenizer
|
|
| 83 |
BATCH_SIZE = 8
|
| 84 |
|
| 85 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 86 |
-
model = AutoModel.from_pretrained('
|
| 87 |
model = model.to_bettertransformer() # Optional: convert the model into a BetterTransformer
|
| 88 |
# to speed it up.
|
| 89 |
-
tokeniser = AutoTokenizer.from_pretrained('
|
| 90 |
|
| 91 |
texts = [
|
| 92 |
'The Parliament shall, subject to this Constitution,\
|
|
|
|
| 83 |
BATCH_SIZE = 8
|
| 84 |
|
| 85 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 86 |
+
model = AutoModel.from_pretrained('umarbutler/emubert').to(device)
|
| 87 |
model = model.to_bettertransformer() # Optional: convert the model into a BetterTransformer
|
| 88 |
# to speed it up.
|
| 89 |
+
tokeniser = AutoTokenizer.from_pretrained('umarbutler/emubert')
|
| 90 |
|
| 91 |
texts = [
|
| 92 |
'The Parliament shall, subject to this Constitution,\
|