Safetensors
xalma
haoranxu commited on
Commit
7e72151
·
verified ·
1 Parent(s): df3cb11

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -127,8 +127,8 @@ input_ids = tokenizer(prompt, return_tensors="pt", padding=True, max_length=40,
127
 
128
  # Translation
129
  with torch.no_grad():
130
- generated_ids = model.generate(input_ids=input_ids, num_beams=5, max_new_tokens=20, do_sample=True, temperature=0.6, top_p=0.9)
131
- outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
132
  print(outputs)
133
  ```
134
 
 
127
 
128
  # Translation
129
  with torch.no_grad():
130
+ generated_ids = model.generate(input_ids=input_ids, num_beams=5, max_new_tokens=20, do_sample=True, temperature=0.6, top_p=0.9)
131
+ outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
132
  print(outputs)
133
  ```
134