Upload T5ForConditionalGeneration
Browse files- README.md +10 -12
- config.json +1 -1
- model.safetensors +1 -1
README.md
CHANGED
@@ -1,22 +1,20 @@
|
|
1 |
---
|
2 |
-
|
|
|
3 |
license: apache-2.0
|
|
|
4 |
datasets:
|
5 |
- ms_marco
|
6 |
-
language:
|
7 |
-
- en
|
8 |
pipeline_tag: text2text-generation
|
9 |
-
|
10 |
widget:
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
inference_config:
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
---
|
21 |
|
22 |
## Model Summary
|
|
|
1 |
---
|
2 |
+
language:
|
3 |
+
- en
|
4 |
license: apache-2.0
|
5 |
+
library_name: transformers
|
6 |
datasets:
|
7 |
- ms_marco
|
|
|
|
|
8 |
pipeline_tag: text2text-generation
|
|
|
9 |
widget:
|
10 |
+
- text: how to bake perfect cookie
|
11 |
+
pipeline_tag: text2text-generation
|
|
|
12 |
inference_config:
|
13 |
+
generation_config:
|
14 |
+
max_length: 35
|
15 |
+
num_beams: 1
|
16 |
+
do_sample: true
|
17 |
+
repetition_penalty: 1.8
|
18 |
---
|
19 |
|
20 |
## Model Summary
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/home/ubuntu/QueryReformulationRL/models/
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "/home/ubuntu/QueryReformulationRL/models/QueryReformulationRLRepPenalty_300",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 891644712
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c53c11b35861cca585ab3f46f5472a8f3b27e8135ecc959165600eee5b5a2271
|
3 |
size 891644712
|