ylacombe HF staff commited on
Commit
85e3762
1 Parent(s): 5cc3f82

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "/fsx/yoach/tmp/artefacts/training-400M-punctuated-v2/",
3
  "architectures": [
4
- "StableSpeechForConditionalGeneration"
5
  ],
6
  "audio_encoder": {
7
  "_name_or_path": "ylacombe/dac_44khZ_8kbps",
@@ -79,7 +79,7 @@
79
  "activation_function": "gelu",
80
  "add_cross_attention": true,
81
  "architectures": [
82
- "StableSpeechForCausalLM"
83
  ],
84
  "attention_dropout": 0.0,
85
  "bad_words_ids": null,
@@ -116,7 +116,7 @@
116
  "max_length": 20,
117
  "max_position_embeddings": 4096,
118
  "min_length": 0,
119
- "model_type": "stable_speech_decoder",
120
  "no_repeat_ngram_size": 0,
121
  "num_attention_heads": 16,
122
  "num_beam_groups": 1,
@@ -155,7 +155,7 @@
155
  },
156
  "decoder_start_token_id": 1025,
157
  "is_encoder_decoder": true,
158
- "model_type": "stable_speech",
159
  "pad_token_id": 1024,
160
  "text_encoder": {
161
  "_name_or_path": "google/flan-t5-base",
 
1
  {
2
  "_name_or_path": "/fsx/yoach/tmp/artefacts/training-400M-punctuated-v2/",
3
  "architectures": [
4
+ "ParlerTTSForConditionalGeneration"
5
  ],
6
  "audio_encoder": {
7
  "_name_or_path": "ylacombe/dac_44khZ_8kbps",
 
79
  "activation_function": "gelu",
80
  "add_cross_attention": true,
81
  "architectures": [
82
+ "ParlerTTSForCausalLM"
83
  ],
84
  "attention_dropout": 0.0,
85
  "bad_words_ids": null,
 
116
  "max_length": 20,
117
  "max_position_embeddings": 4096,
118
  "min_length": 0,
119
+ "model_type": "parler_tts_decoder",
120
  "no_repeat_ngram_size": 0,
121
  "num_attention_heads": 16,
122
  "num_beam_groups": 1,
 
155
  },
156
  "decoder_start_token_id": 1025,
157
  "is_encoder_decoder": true,
158
+ "model_type": "parler_tts",
159
  "pad_token_id": 1024,
160
  "text_encoder": {
161
  "_name_or_path": "google/flan-t5-base",