dar-tau commited on
Commit
724d551
1 Parent(s): df5ce56

Upload ToyAutoEncoder

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -1,9 +1,10 @@
1
  {
2
- "_name_or_path": "dar-tau/toy-autoencoder-4L-1L",
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",