flavioschneider commited on
Commit
0d8c061
·
1 Parent(s): 10c5045

Upload config

Browse files
Files changed (2) hide show
  1. autoencoder_config.py +27 -0
  2. config.json +36 -0
autoencoder_config.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PretrainedConfig
2
+ from typing import Sequence
3
+
4
+ class AutoEncoder1dConfig(PretrainedConfig):
5
+
6
+ model_type = "archinetai/autoencoder1d-AT-v1"
7
+
8
+ def __init__(
9
+ self,
10
+ in_channels: int = 2,
11
+ patch_size: int = 4,
12
+ channels: int = 32,
13
+ multipliers: Sequence[int] = [1, 2, 4, 8, 8, 8, 1],
14
+ factors: Sequence[int] = [2, 2, 2, 1, 1, 1],
15
+ num_blocks: Sequence[int] = [2, 2, 8, 8, 8, 8],
16
+ bottleneck: str = 'tanh',
17
+ **kwargs
18
+ ):
19
+ self.in_channels = in_channels
20
+ self.patch_size = patch_size
21
+ self.channels = channels
22
+ self.multipliers = multipliers
23
+ self.factors = factors
24
+ self.num_blocks = num_blocks
25
+ self.bottleneck = bottleneck
26
+ super().__init__(**kwargs)
27
+
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoConfig": "autoencoder_config.AutoEncoder1dConfig"
4
+ },
5
+ "bottleneck": "tanh",
6
+ "channels": 32,
7
+ "factors": [
8
+ 2,
9
+ 2,
10
+ 2,
11
+ 1,
12
+ 1,
13
+ 1
14
+ ],
15
+ "in_channels": 2,
16
+ "model_type": "archinetai/autoencoder1d-AT-v1",
17
+ "multipliers": [
18
+ 1,
19
+ 2,
20
+ 4,
21
+ 8,
22
+ 8,
23
+ 8,
24
+ 1
25
+ ],
26
+ "num_blocks": [
27
+ 2,
28
+ 2,
29
+ 8,
30
+ 8,
31
+ 8,
32
+ 8
33
+ ],
34
+ "patch_size": 4,
35
+ "transformers_version": "4.24.0"
36
+ }