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
|