elmadany commited on
Commit
fee5714
·
verified ·
1 Parent(s): df53f0d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -178,12 +178,12 @@ text="zul: Clear all items from the recent documents list"
178
  input_ids = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True).to("cuda:0")
179
  with torch.no_grad():
180
  generated_ids = model.generate(**input_ids, num_beams=5, max_new_tokens=len(text), do_sample=True, temperature=0.6, top_p=0.9)
181
- print(tokenizer.batch_decode(generated_ids, skip_special_tokens=True, skip_prompt=True)[0])
182
 
183
  ```
184
  Output:
185
  ```bash
186
- Vala zonke izinto kusuka kwihlu lamadokhumende elidlule
187
  ```
188
 
189
 
 
178
  input_ids = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True).to("cuda:0")
179
  with torch.no_grad():
180
  generated_ids = model.generate(**input_ids, num_beams=5, max_new_tokens=len(text), do_sample=True, temperature=0.6, top_p=0.9)
181
+ print("Toucan-base - translation:", tokenizer.batch_decode(generated_ids, skip_special_tokens=True, skip_prompt=True)[0])
182
 
183
  ```
184
  Output:
185
  ```bash
186
+ Toucan-base - translation: Vala zonke izinto kusuka kwihlu lamadokhumende elidlule
187
  ```
188
 
189