drixo commited on
Commit
2a7db15
·
verified ·
1 Parent(s): 3094fbb

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +7 -8
config.py CHANGED
@@ -1,9 +1,8 @@
1
  class TTSConfig:
2
- vocab_size = 8000
3
- d_model = 256
4
- n_heads = 4
5
- num_encoder_layers = 4
6
- num_decoder_layers = 4
7
- mel_bins = 80
8
- max_seq_len = 2048
9
- sample_rate = 22050
 
1
  class TTSConfig:
2
+ def __init__(self):
3
+ self.vocab_size = 5000
4
+ self.d_model = 256
5
+ self.nhead = 4
6
+ self.num_layers = 4
7
+ self.dim_feedforward = 512
8
+ self.max_seq_len = 512