mstatt commited on
Commit
b907573
1 Parent(s): 2ec3228

Upload T5ForConditionalGeneration

Browse files
Files changed (3) hide show
  1. config.json +21 -3
  2. generation_config.json +0 -6
  3. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "Falconsai/medical_summarization",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
@@ -28,13 +28,31 @@
28
  "summarization": {
29
  "early_stopping": true,
30
  "length_penalty": 2.0,
31
- "max_length": 300,
32
  "min_length": 30,
33
  "no_repeat_ngram_size": 3,
34
  "num_beams": 4,
35
  "prefix": "summarize: "
36
  },
37
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38
  "torch_dtype": "float32",
39
  "transformers_version": "4.31.0",
40
  "use_cache": true,
 
1
  {
2
+ "_name_or_path": "./model/",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
 
28
  "summarization": {
29
  "early_stopping": true,
30
  "length_penalty": 2.0,
31
+ "max_length": 200,
32
  "min_length": 30,
33
  "no_repeat_ngram_size": 3,
34
  "num_beams": 4,
35
  "prefix": "summarize: "
36
  },
37
+ "translation_en_to_de": {
38
+ "early_stopping": true,
39
+ "max_length": 300,
40
+ "num_beams": 4,
41
+ "prefix": "translate English to German: "
42
+ },
43
+ "translation_en_to_fr": {
44
+ "early_stopping": true,
45
+ "max_length": 300,
46
+ "num_beams": 4,
47
+ "prefix": "translate English to French: "
48
+ },
49
+ "translation_en_to_ro": {
50
+ "early_stopping": true,
51
+ "max_length": 300,
52
+ "num_beams": 4,
53
+ "prefix": "translate English to Romanian: "
54
+ }
55
+ },
56
  "torch_dtype": "float32",
57
  "transformers_version": "4.31.0",
58
  "use_cache": true,
generation_config.json CHANGED
@@ -1,12 +1,6 @@
1
  {
2
  "decoder_start_token_id": 0,
3
- "early_stopping": true,
4
  "eos_token_id": 1,
5
- "length_penalty": 2.0,
6
- "max_length": 200,
7
- "min_length": 30,
8
- "no_repeat_ngram_size": 3,
9
- "num_beams": 4,
10
  "pad_token_id": 0,
11
  "transformers_version": "4.31.0"
12
  }
 
1
  {
2
  "decoder_start_token_id": 0,
 
3
  "eos_token_id": 1,
 
 
 
 
 
4
  "pad_token_id": 0,
5
  "transformers_version": "4.31.0"
6
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c806f975ded2d8f21869ada8534e79b72555c0947ee3aef12facf2aeed9a260
3
  size 242069785
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1a6a370921cde2862afd981d62d6d6a06ffcf6739ca691d02d971e455915168
3
  size 242069785