manu commited on
Commit
434580e
·
verified ·
1 Parent(s): 5f24e3c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -60,4 +60,5 @@ print(tokenizer.decode(tokens[0]))
60
  # remove bos token
61
  inputs = tokenizer("Capitales: France -> Paris, Italie -> Rome, Allemagne -> Berlin, Espagne ->", return_tensors="pt", add_special_tokens=True).to(model.device)
62
  tokens = model.generate(**inputs, max_length=100, do_sample=True, top_p=0.95, top_k=60)
63
- print(tokenizer.decode(tokens[0]))
 
 
60
  # remove bos token
61
  inputs = tokenizer("Capitales: France -> Paris, Italie -> Rome, Allemagne -> Berlin, Espagne ->", return_tensors="pt", add_special_tokens=True).to(model.device)
62
  tokens = model.generate(**inputs, max_length=100, do_sample=True, top_p=0.95, top_k=60)
63
+ print(tokenizer.decode(tokens[0]))
64
+ ```