Update README.md
Browse files
README.md
CHANGED
|
@@ -163,7 +163,7 @@ input_text = """
|
|
| 163 |
|
| 164 |
SELECT"""
|
| 165 |
|
| 166 |
-
encoding = tokenizer.encode_plus(input_text,
|
| 167 |
input_ids, attention_masks = encoding["input_ids"].to(device), encoding["attention_mask"].to(device)
|
| 168 |
|
| 169 |
|
|
@@ -233,7 +233,7 @@ input_text = """
|
|
| 233 |
SELECT"""
|
| 234 |
|
| 235 |
|
| 236 |
-
encoding = tokenizer.encode_plus(input_text,
|
| 237 |
input_ids, attention_masks = encoding["input_ids"].to(device), encoding["attention_mask"].to(device)
|
| 238 |
|
| 239 |
|
|
|
|
| 163 |
|
| 164 |
SELECT"""
|
| 165 |
|
| 166 |
+
encoding = tokenizer.encode_plus(input_text, return_tensors="pt").to(device)
|
| 167 |
input_ids, attention_masks = encoding["input_ids"].to(device), encoding["attention_mask"].to(device)
|
| 168 |
|
| 169 |
|
|
|
|
| 233 |
SELECT"""
|
| 234 |
|
| 235 |
|
| 236 |
+
encoding = tokenizer.encode_plus(input_text, return_tensors="pt").to(device)
|
| 237 |
input_ids, attention_masks = encoding["input_ids"].to(device), encoding["attention_mask"].to(device)
|
| 238 |
|
| 239 |
|