Summer11t
/

summer.lab

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Summer11t commited on May 13

Commit

4f039d0

•

1 Parent(s): ee38dfc

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -12,8 +12,8 @@
   "model_type": "gpt2",
   "n_ctx": 1024,
   "n_embd": 768,
-  "n_head": 32,
-  "n_layer": 32,
   "n_positions": 1024,
   "resid_pdrop": 0.1,
   "summary_activation": null,
@@ -24,7 +24,7 @@
   "task_specific_params": {
     "text-generation": {
       "do_sample": true,
-      "max_length": 3000
     }
   },
   "vocab_size": 50257

   "model_type": "gpt2",
   "n_ctx": 1024,
   "n_embd": 768,
+  "n_head": 12,
+  "n_layer": 12,
   "n_positions": 1024,
   "resid_pdrop": 0.1,
   "summary_activation": null,
   "task_specific_params": {
     "text-generation": {
       "do_sample": true,
+      "max_length": 200
     }
   },
   "vocab_size": 50257