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 |
|