Fix typos in model card
Browse files
README.md
CHANGED
|
@@ -58,9 +58,9 @@ prompt = f"""### Task Description
|
|
| 58 |
{statement}
|
| 59 |
|
| 60 |
### Factual
|
| 61 |
-
"""
|
| 62 |
|
| 63 |
-
inputs = tokenizer([prompt], padding=True, truncation=True,
|
| 64 |
inputs = {k: v.to(model.device) for k, v in inputs.items()}
|
| 65 |
|
| 66 |
results = model.generate(**inputs, max_new_tokens=128)
|
|
|
|
| 58 |
{statement}
|
| 59 |
|
| 60 |
### Factual
|
| 61 |
+
"""
|
| 62 |
|
| 63 |
+
inputs = tokenizer([prompt], padding=True, truncation=True, return_tensors='pt')
|
| 64 |
inputs = {k: v.to(model.device) for k, v in inputs.items()}
|
| 65 |
|
| 66 |
results = model.generate(**inputs, max_new_tokens=128)
|