Dmitry Chaplinsky
commited on
Commit
•
320a61b
1
Parent(s):
30f9897
Decreasing the heat a bit
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -19,4 +19,4 @@ class PreTrainedPipeline:
|
|
19 |
A :obj:`str`
|
20 |
"""
|
21 |
inputs = inputs.strip()
|
22 |
-
return [{"generated_text": self.model.generate_text(inputs)[0]}]
|
|
|
19 |
A :obj:`str`
|
20 |
"""
|
21 |
inputs = inputs.strip()
|
22 |
+
return [{"generated_text": self.model.generate_text(inputs, temperature=0.25)[0]}]
|