ArtifactAI commited on
Commit
f6f0fa1
·
1 Parent(s): d221e4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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:\n{tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0]}")
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