Zane
/

Ricky

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

Zane commited on Jul 29, 2021

Commit

af56732

•

1 Parent(s): 3a8aa26

Update config.json

Files changed (1) hide show

config.json +3 -4

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "microsoft/DialoGPT-small",
   "activation_function": "gelu_new",
   "architectures": [
     "GPT2LMHeadModel"
@@ -30,8 +30,7 @@
       "max_length": 1000
     }
   },
-  "torch_dtype": "float32",
-  "transformers_version": "4.9.1",
   "use_cache": true,
   "vocab_size": 50257
-}

 {
+  "_name_or_path": "output-small-neku",
   "activation_function": "gelu_new",
   "architectures": [
     "GPT2LMHeadModel"
       "max_length": 1000
     }
   },
+  "transformers_version": "4.6.1",
   "use_cache": true,
   "vocab_size": 50257
+}