Upload 2 files

config.json

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "LightningSpeechModel"
+    "LightningSpeechV2Model"
   ],
   "bos_token_id": 1,
   "encoder_config": {
@@ -23,27 +23,10 @@
     11,
     13
   ],
-  "mask_time_length": 10,
-  "mask_time_min_masks": 2,
-  "mask_time_prob": 0.65,
-  "max_prompt_length_ratio": 0.5,
-  "min_prompt_length_ratio": 0.1,
-  "model_type": "lightning_speech",
+  "model_type": "lightning_speech_v2",
   "num_semantic_tokens": 1024,
   "pad_token_id": 0,
-  "prosody_predictor_config": {
-    "attn_dropout": 0.0,
-    "attn_flash": true,
-    "conv_dropout": 0.0,
-    "conv_kernel_size": 5,
-    "cross_attention": true,
-    "depth": 2,
-    "dim_head": 64,
-    "ff_dropout": 0.0,
-    "ff_mult": 4,
-    "heads": 16
-  },
   "quantizer_drop_prob": 0.75,
   "torch_dtype": "bfloat16",
-  "transformers_version": "4.38.0.dev0"
+  "transformers_version": "4.42.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c487cc944e594c2a25afb1a3ff935f38d9a31a39a0a9fc90b27398c612e62521
-size 1204891232
+oid sha256:c2f16495f222bd2f33856882917d357d75082ae83b6b9345008cdf727e23f3a4
+size 1076777908