Update README.md
Browse files
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 |
+
```
|