elmadany commited on
Commit
c640f2c
·
verified ·
1 Parent(s): cb366c3

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-1.2B - translation:", tokenizer.batch_decode(generated_ids, skip_special_tokens=True, skip_prompt=True)[0])
182
 
183
  ```
184
  Output:
185
  ```bash
186
+ Toucan-1.2B - translation: Susa zonke izinto kuhlu lwamadokhumende oludlule
187
  ```
188
 
189