Update config.json

config.json

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "/fsx/yoach/tmp/artefacts/training-400M-punctuated-v2/",
   "architectures": [
-    "StableSpeechForConditionalGeneration"
+    "ParlerTTSForConditionalGeneration"
   ],
   "audio_encoder": {
     "_name_or_path": "ylacombe/dac_44khZ_8kbps",
@@ -79,7 +79,7 @@
     "activation_function": "gelu",
     "add_cross_attention": true,
     "architectures": [
-      "StableSpeechForCausalLM"
+      "ParlerTTSForCausalLM"
     ],
     "attention_dropout": 0.0,
     "bad_words_ids": null,
@@ -116,7 +116,7 @@
     "max_length": 20,
     "max_position_embeddings": 4096,
     "min_length": 0,
-    "model_type": "stable_speech_decoder",
+    "model_type": "parler_tts_decoder",
     "no_repeat_ngram_size": 0,
     "num_attention_heads": 16,
     "num_beam_groups": 1,
@@ -155,7 +155,7 @@
   },
   "decoder_start_token_id": 1025,
   "is_encoder_decoder": true,
-  "model_type": "stable_speech",
+  "model_type": "parler_tts",
   "pad_token_id": 1024,
   "text_encoder": {
     "_name_or_path": "google/flan-t5-base",