Create inference.yaml
Browse files- inference.yaml +8 -0
inference.yaml
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
pipeline_type: text-generation
|
| 2 |
+
|
| 3 |
+
default_parameters:
|
| 4 |
+
max_length: 150
|
| 5 |
+
num_return_sequences: 1
|
| 6 |
+
temperature: 0.3
|
| 7 |
+
top_k: 50
|
| 8 |
+
top_p: 0.95
|