Update README.md
Browse files
README.md
CHANGED
|
@@ -44,6 +44,6 @@ tokenizer = AutoTokenizer.from_pretrained("MaxLSB/LeCarnet-3M")
|
|
| 44 |
prompt = "Il était une fois"
|
| 45 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
| 46 |
|
| 47 |
-
output = model.generate(input_ids, max_length=512)
|
| 48 |
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
| 49 |
```
|
|
|
|
| 44 |
prompt = "Il était une fois"
|
| 45 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
| 46 |
|
| 47 |
+
output = model.generate(input_ids, max_length=512, temperature=0.4)
|
| 48 |
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
| 49 |
```
|