ArtifactAI
commited on
Commit
·
f6f0fa1
1
Parent(s):
d221e4f
Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ input_ids = tokenizer("What is the peak phase of T-eV?", return_tensors="pt", tr
|
|
| 59 |
# with torch.inference_mode():
|
| 60 |
outputs = model.generate(input_ids=input_ids, max_new_tokens=1000, do_sample=True, top_p=0.9)
|
| 61 |
|
| 62 |
-
print(f"summary
|
| 63 |
```
|
| 64 |
|
| 65 |
## Training Data
|
|
|
|
| 59 |
# with torch.inference_mode():
|
| 60 |
outputs = model.generate(input_ids=input_ids, max_new_tokens=1000, do_sample=True, top_p=0.9)
|
| 61 |
|
| 62 |
+
print(f"summary: {tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0]}")
|
| 63 |
```
|
| 64 |
|
| 65 |
## Training Data
|