Upload ToyAutoEncoder
Browse files- config.json +2 -1
config.json
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "dar-tau/toy-autoencoder-
|
3 |
"architectures": [
|
4 |
"ToyAutoEncoder"
|
5 |
],
|
6 |
"decoder_specs": {
|
|
|
7 |
"model_type": "",
|
8 |
"residual": true,
|
9 |
"type": "mlp",
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "dar-tau/toy-autoencoder-mlp",
|
3 |
"architectures": [
|
4 |
"ToyAutoEncoder"
|
5 |
],
|
6 |
"decoder_specs": {
|
7 |
+
"activation": "gelu",
|
8 |
"model_type": "",
|
9 |
"residual": true,
|
10 |
"type": "mlp",
|