Wonder-Griffin
/

Judge-GPT2

Generated from Trainer

Model card Files Files and versions Metrics Training metrics Community

Wonder-Griffin commited on Aug 15, 2024

Commit

b771bee

·

verified ·

1 Parent(s): 9a97de5

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "GPT2LMHeadModel"
   ],
   "inference_mode": true,
   "batch_size": 32,
@@ -60,7 +60,7 @@
   "task_specific_params": {
     "text-generation": {
       "do_sample": true,
-      "max_length": 50
     }
   },
   "task_type": "SEQ_CLS",

 {
   "architectures": [
+    "openai-gpt"
   ],
   "inference_mode": true,
   "batch_size": 32,
   "task_specific_params": {
     "text-generation": {
       "do_sample": true,
+      "max_length": 100
     }
   },
   "task_type": "SEQ_CLS",