kmfoda
/

staging-pegasus-gmeetsamsum

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

kmfoda commited on Feb 2, 2022

Commit

835c702

•

1 Parent(s): a0f1bb7

Update config.json

Files changed (1) hide show

config.json +4 -3

config.json CHANGED Viewed

@@ -1,11 +1,11 @@
 {
-  "_name_or_path": "google/pegasus-large",
   "activation_dropout": 0.1,
   "activation_function": "relu",
   "add_bias_logits": false,
   "add_final_layer_norm": true,
   "architectures": [
-    "PegasusForConditionalGeneration"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 0,
@@ -117,7 +117,8 @@
       "max_position_embeddings": 512
     }
   },
-  "transformers_version": "4.11.0.dev0",
   "use_cache": true,
   "vocab_size": 96103
 }

 {
+  "_name_or_path": "../output_dir/",
   "activation_dropout": 0.1,
   "activation_function": "relu",
   "add_bias_logits": false,
   "add_final_layer_norm": true,
   "architectures": [
+    "PegasusModel"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 0,
       "max_position_embeddings": 512
     }
   },
+  "torch_dtype": "float32",
+  "transformers_version": "4.16.0.dev0",
   "use_cache": true,
   "vocab_size": 96103
 }