umarbutler
commited on
Commit
•
1afde0c
1
Parent(s):
5eb8531
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,\
|