Commit ·
7646c36
1
Parent(s): 3125d16
Update README.md
Browse files
README.md
CHANGED
|
@@ -39,11 +39,7 @@ generation_pipeline = pipeline("text-generation",model=model,tokenizer=tokenizer
|
|
| 39 |
#feel free to try different decoding settings
|
| 40 |
generation_pipeline(text,
|
| 41 |
pad_token_id=tokenizer.eos_token_id,
|
| 42 |
-
|
| 43 |
-
max_length=512,
|
| 44 |
penalty_alpha=0.6,
|
| 45 |
-
|
| 46 |
-
top_k=4,
|
| 47 |
-
repetition_penalty = 3.0,
|
| 48 |
-
no_repeat_ngram_size = 3)[0]['generated_text']
|
| 49 |
```
|
|
|
|
| 39 |
#feel free to try different decoding settings
|
| 40 |
generation_pipeline(text,
|
| 41 |
pad_token_id=tokenizer.eos_token_id,
|
| 42 |
+
max_length=512,
|
|
|
|
| 43 |
penalty_alpha=0.6,
|
| 44 |
+
top_k=4 )[0]['generated_text']
|
|
|
|
|
|
|
|
|
|
| 45 |
```
|