Update README.md
Browse files
README.md
CHANGED
@@ -73,10 +73,10 @@ outputs = pipeline(
|
|
73 |
prompt,
|
74 |
max_new_tokens=2048,
|
75 |
eos_token_id=terminators,
|
76 |
-
do_sample=
|
77 |
temperature=0.6,
|
78 |
top_p=1,
|
79 |
-
repetition_penalty=1.
|
80 |
)
|
81 |
print(outputs[0]["generated_text"][len(prompt):])
|
82 |
```
|
|
|
73 |
prompt,
|
74 |
max_new_tokens=2048,
|
75 |
eos_token_id=terminators,
|
76 |
+
do_sample=True,
|
77 |
temperature=0.6,
|
78 |
top_p=1,
|
79 |
+
repetition_penalty=1.15
|
80 |
)
|
81 |
print(outputs[0]["generated_text"][len(prompt):])
|
82 |
```
|