Update config.json
Browse files- config.json +1 -4
config.json
CHANGED
@@ -1,7 +1,4 @@
|
|
1 |
{
|
2 |
-
{
|
3 |
"model_type": "gpt2",
|
4 |
-
"
|
5 |
-
"hidden_size": 384,
|
6 |
-
"intermediate_size": 1024
|
7 |
}
|
|
|
1 |
{
|
|
|
2 |
"model_type": "gpt2",
|
3 |
+
"pipeline_tag": "text-generation"
|
|
|
|
|
4 |
}
|