HiTeckGroup prabhacar007 commited on
Commit
79effea
·
0 Parent(s):

Duplicate from voxaura-labs/vaakya-open

Browse files

Co-authored-by: prabhakar srinivasan <prabhacar007@users.noreply.huggingface.co>

.gitattributes ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ samples/final_test_1_english.wav filter=lfs diff=lfs merge=lfs -text
37
+ samples/final_test_2_hindi.wav filter=lfs diff=lfs merge=lfs -text
38
+ samples/final_test_3_code-mixed.wav filter=lfs diff=lfs merge=lfs -text
39
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
40
+ final_test_1_english.wav filter=lfs diff=lfs merge=lfs -text
41
+ final_test_2_hindi.wav filter=lfs diff=lfs merge=lfs -text
42
+ final_test_3_code-mixed.wav filter=lfs diff=lfs merge=lfs -text
43
+ weather.wav filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,616 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - TTS
5
+ - Hinglish
6
+ - Indian-languages
7
+ - Voice-AI
8
+ - Speech-Synthesis
9
+ language:
10
+ - hi
11
+ - en
12
+ base_model:
13
+ - canopylabs/orpheus-3b-0.1-ft
14
+ pipeline_tag: text-to-speech
15
+ ---
16
+ # Vaakya‑Open
17
+
18
+ **Vaakya‑Open** is a high‑quality, single‑speaker **Text‑to‑Speech (TTS)** model developed by **Voxaura Labs**, designed for **English and Hindi** voice synthesis. It features a **natural female voice**, optimized for voice‑overs, audiobooks, podcasts, narration, assistants, and production‑grade applications.
19
+
20
+ Built with a strong focus on clarity, consistency, and expressiveness, Vaakya‑Open is ideal for creators and developers looking for a dependable, studio‑like voice that works seamlessly across English, Hindi, and code‑mixed inputs.
21
+
22
+ ---
23
+
24
+ ## 🔊 Live Demo
25
+
26
+ 👉 **Try it instantly** using the accompanying **Gradio Space**:
27
+
28
+ > **[Vaakya‑Open TTS Demo](https://huggingface.co/spaces/voxaura-labs/vaakya-open-demo)** — Convert English or Hindi text into natural‑sounding speech directly in your browser.
29
+
30
+ Note! If the space has become dormant due to inactivity, you may restart the space. Restart takes about 4 minutes (T4 small VM)
31
+
32
+ ---
33
+
34
+ ## 🎧 Audio Samples
35
+
36
+ Listen to sample outputs demonstrating the model's capabilities:
37
+
38
+ | Sample | Language | Audio Player |
39
+ |--------|----------|--------------|
40
+ | **English** | Pure English | <audio controls src="https://huggingface.co/voxaura-labs/vaakya-open/resolve/main/final_test_1_english.wav"></audio> |
41
+ | **Hindi** | Pure Hindi (Devanagari) | <audio controls src="https://huggingface.co/voxaura-labs/vaakya-open/resolve/main/final_test_2_hindi.wav"></audio> |
42
+ | **Code‑Mixed** | Hindi + English (Hinglish) | <audio controls src="https://huggingface.co/voxaura-labs/vaakya-open/resolve/main/weather.wav"></audio> |
43
+
44
+ All samples are generated at 24kHz with 16-bit PCM encoding.
45
+
46
+ ---
47
+
48
+ ## ✨ Key Highlights
49
+
50
+ - 🎙️ **Single Professional Female Voice** — consistent, warm, and narration‑ready
51
+ - 🌐 **Bilingual Support** — English & Hindi (with natural code‑mixing)
52
+ - 🎧 **Studio‑Quality Audio** — trained on pristine 192kHz recordings, output at 24kHz
53
+ - ⚡ **Low‑Latency Inference** — suitable for real‑time and batch workflows
54
+ - 🧠 **Production‑Oriented** — stable voice characteristics across long passages
55
+
56
+ ---
57
+
58
+ ## 🧠 Model Overview
59
+
60
+ | Attribute | Details |
61
+ |-----------|---------|
62
+ | **Model Name** | Vaakya‑Open |
63
+ | **Model Type** | Autoregressive Transformer (Speech‑LLM) |
64
+ | **Base Architecture** | Llama 3B |
65
+ | **Speaker** | Single speaker (Female) |
66
+ | **Languages** | English, Hindi, Code‑mixed |
67
+ | **Audio Codec** | SNAC @ 24kHz |
68
+ | **Sampling Rate** | 24 kHz (output) |
69
+ | **Developed By** | Voxaura Labs |
70
+ | **License** | Apache 2.0 |
71
+
72
+ ---
73
+
74
+ ## 🏗️ Architecture
75
+
76
+ Vaakya‑Open is built on the **Orpheus TTS** architecture pioneered by [Canopy Labs](https://canopylabs.ai), which treats speech synthesis as a language modeling task. The model generates discrete audio tokens that are decoded into high‑quality waveforms.
77
+
78
+ ```
79
+ ┌──────────────────────────────────────────────────────────────────────────┐
80
+ │ VAAKYA-OPEN TTS ARCHITECTURE │
81
+ └──────────────────────────────────────────────────────────────────────────┘
82
+
83
+ ╔══════════════════╗
84
+ ║ Text Input ║ English / Hindi / Code-Mixed
85
+ ║ ║
86
+ ╚════════╤═════════╝
87
+ │
88
+ ▼
89
+ ╔═══════════════════════════╗
90
+ ║ Llama 3B LLM ║ Autoregressive generation
91
+ ║ (Speech Transformer) ║ 7 tokens per audio frame
92
+ ╚═════════════╤═════════════╝
93
+ │
94
+ ▼
95
+ ╔═══════════════════════════╗
96
+ ║ Audio Tokens ║ Discrete codes
97
+ ║ (SNAC Format) ║ Hierarchical 3-level
98
+ ╚═════════════╤═════════════╝
99
+ │
100
+ ▼
101
+ ╔═══════════════════════════╗
102
+ ║ SNAC Decoder ║ Neural audio codec
103
+ ║ (24kHz) ║ Token → Waveform
104
+ ╚═════════════╤═════════════╝
105
+ │
106
+ ▼
107
+ ╔═══════════════════════════╗
108
+ ║ 24kHz Audio Waveform ║ High-quality speech output
109
+ ║ (Output) ║ Studio-grade quality
110
+ ╚═══════════════════════════╝
111
+ ```
112
+
113
+ ### How It Works
114
+
115
+ 1. **Text Input** — Your text (English, Hindi, or code‑mixed) is tokenized using a text tokenizer
116
+ 2. **Audio Token Generation** — The Llama‑based LLM autoregressively generates discrete audio tokens (7 tokens per audio frame)
117
+ 3. **SNAC Decoding** — The [SNAC neural codec](https://github.com/hubertsiuzdak/snac) converts audio tokens back into a 24kHz waveform
118
+ 4. **Output** — High‑quality speech audio ready for playback or further processing
119
+
120
+ ### Key Architectural Features
121
+
122
+ | Component | Specification |
123
+ |-----------|---------------|
124
+ | **LLM Backbone** | Llama‑style autoregressive transformer (3B parameters) |
125
+ | **Audio Tokenizer** | SNAC with 7 tokens per frame (flattened sequence) |
126
+ | **Tokens per Second** | ~83 audio tokens/second |
127
+ | **Context Length** | 2048 tokens |
128
+ | **Streaming Support** | Yes (via sliding window decoding) |
129
+
130
+ > **Attribution:** This architecture builds on [Orpheus TTS](https://github.com/canopyai/Orpheus-TTS) by [Canopy Labs](https://canopylabs.ai), which demonstrated that LLMs can achieve human‑level speech synthesis.
131
+
132
+ ---
133
+
134
+ ## 🎼 Voice & Audio Quality
135
+
136
+ | Attribute | Value |
137
+ |-----------|-------|
138
+ | **Original Recording Rate** | 192 kHz (studio‑grade) |
139
+ | **Training / Output Rate** | 24 kHz |
140
+ | **Output Format** | WAVE (PCM 16-bit, mono, 24kHz) |
141
+ | **Recording Environment** | Controlled studio conditions |
142
+ | **Voice Style** | Neutral, professional, voice‑over friendly |
143
+
144
+ This high‑resolution capture pipeline preserves subtle vocal textures, resulting in clean pronunciation, smooth prosody, and reduced artifacts.
145
+
146
+ ---
147
+
148
+ ## 🚀 Getting Started
149
+
150
+ ### Installation
151
+
152
+ ```bash
153
+ pip install torch transformers soundfile accelerate
154
+ pip install snac # For audio decoding
155
+ ```
156
+
157
+ For optional 4-bit quantization support:
158
+ ```bash
159
+ pip install bitsandbytes
160
+ ```
161
+
162
+ ### Basic Usage
163
+
164
+ ```python
165
+ import torch
166
+ import soundfile as sf
167
+ from transformers import AutoTokenizer, AutoModel
168
+
169
+ model_id = "voxaura-labs/vaakya-open"
170
+
171
+ model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
172
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
173
+
174
+ text = "नमस्ते! This is Vaakya‑Open from Voxaura Labs."
175
+
176
+ with torch.no_grad():
177
+ audio = model.generate_speech(text)
178
+
179
+ sf.write("output.wav", audio, 24000)
180
+ ```
181
+
182
+ ### Advanced Usage (Full Pipeline)
183
+
184
+ For users who need more control over generation parameters:
185
+
186
+ ```python
187
+ import torch
188
+ from transformers import AutoModelForCausalLM, AutoTokenizer
189
+ from snac import SNAC
190
+ import soundfile as sf
191
+
192
+ # Load model and tokenizer (FP16 for balanced speed and quality)
193
+ model = AutoModelForCausalLM.from_pretrained(
194
+ "voxaura-labs/vaakya-open",
195
+ torch_dtype=torch.float16,
196
+ device_map="auto",
197
+ trust_remote_code=True,
198
+ )
199
+ tokenizer = AutoTokenizer.from_pretrained("voxaura-labs/vaakya-open")
200
+
201
+ # Initialize SNAC decoder
202
+ # Note: SNAC will be moved to CPU during generation for stability
203
+ snac_model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").eval()
204
+ if torch.cuda.is_available():
205
+ snac_model = snac_model.cuda()
206
+
207
+ # Optional: For faster inference with lower memory usage, use 4-bit quantization:
208
+ # from transformers import BitsAndBytesConfig
209
+ # quantization_config = BitsAndBytesConfig(
210
+ # load_in_4bit=True,
211
+ # bnb_4bit_quant_type="nf4",
212
+ # bnb_4bit_compute_dtype=torch.bfloat16,
213
+ # bnb_4bit_use_double_quant=True,
214
+ # )
215
+ # Then pass quantization_config to from_pretrained() instead of torch_dtype
216
+
217
+ # Token IDs
218
+ START_OF_SPEECH_TOKEN = 128257
219
+ END_OF_SPEECH_TOKEN = 128258
220
+ START_OF_HUMAN_TOKEN = 128259
221
+ END_OF_HUMAN_TOKEN = 128260
222
+ START_OF_AI_TOKEN = 128261
223
+ END_OF_AI_TOKEN = 128262
224
+ AUDIO_CODE_BASE_OFFSET = 128266
225
+
226
+ # Audio token range: 128266 to 156937 (28,672 tokens total)
227
+ # The 7 tokens per frame use offsets: +0, +4096, +8192, +12288, +16384, +20480, +24576
228
+ MAX_AUDIO_TOKEN = 156937
229
+
230
+
231
+ def generate_speech(text, temperature=0.5, top_p=0.9):
232
+ """Generate speech from text.
233
+
234
+ Args:
235
+ text: Input text (English, Hindi, or code-mixed)
236
+ temperature: Sampling temperature (0.4-0.7 recommended)
237
+ top_p: Nucleus sampling parameter
238
+
239
+ Returns:
240
+ numpy array: Audio waveform at 24kHz
241
+ """
242
+ # Move SNAC to CPU for decoding (important for stability)
243
+ snac_model.to("cpu")
244
+
245
+ # Tokenize text (automatically adds BOS token)
246
+ input_ids = tokenizer(text, return_tensors="pt").input_ids
247
+
248
+ # Create input sequence: START_HUMAN + [BOS + Text + EOS] + END_HUMAN
249
+ start_token = torch.tensor([[START_OF_HUMAN_TOKEN]], dtype=torch.int64)
250
+ end_tokens = torch.tensor([[128009, END_OF_HUMAN_TOKEN]], dtype=torch.int64) # EOS + END_HUMAN
251
+
252
+ input_sequence = torch.cat([start_token, input_ids, end_tokens], dim=1)
253
+ input_sequence = input_sequence.to(model.device)
254
+
255
+ # Generate audio tokens
256
+ with torch.no_grad():
257
+ generated_ids = model.generate(
258
+ input_ids=input_sequence,
259
+ max_new_tokens=2048,
260
+ do_sample=True,
261
+ temperature=temperature,
262
+ top_p=top_p,
263
+ repetition_penalty=1.1,
264
+ num_return_sequences=1,
265
+ eos_token_id=END_OF_SPEECH_TOKEN,
266
+ use_cache=True,
267
+ pad_token_id=tokenizer.pad_token_id,
268
+ )
269
+
270
+ # Extract generated tokens
271
+ gen_tokens = generated_ids[0]
272
+
273
+ # Find the last occurrence of START_OF_SPEECH token
274
+ sos_indices = (gen_tokens == START_OF_SPEECH_TOKEN).nonzero(as_tuple=True)[0]
275
+
276
+ if len(sos_indices) > 0:
277
+ # Start from after the last START_OF_SPEECH token
278
+ start_idx = sos_indices[-1].item() + 1
279
+ cropped_tokens = gen_tokens[start_idx:]
280
+ else:
281
+ cropped_tokens = gen_tokens
282
+
283
+ # Remove END_OF_SPEECH tokens
284
+ audio_tokens = cropped_tokens[cropped_tokens != END_OF_SPEECH_TOKEN]
285
+
286
+ # Truncate to make divisible by 7
287
+ num_tokens = len(audio_tokens)
288
+ num_frames = num_tokens // 7
289
+ audio_tokens = audio_tokens[:num_frames * 7]
290
+
291
+ if len(audio_tokens) == 0:
292
+ raise ValueError("No audio tokens generated")
293
+
294
+ # Convert to list and subtract offset
295
+ code_list = [t.item() - AUDIO_CODE_BASE_OFFSET for t in audio_tokens]
296
+
297
+ # Decode to audio
298
+ return decode_snac_tokens(code_list)
299
+
300
+
301
+ def decode_snac_tokens(code_list):
302
+ """Decode SNAC tokens to audio waveform.
303
+
304
+ This model uses a 7-token interleaved encoding per audio frame with offsets:
305
+ Token 1: offset +0 (level 0, coarse)
306
+ Token 2: offset +4096 (level 1, medium)
307
+ Token 3: offset +8192 (level 2, fine)
308
+ Token 4: offset +12288 (level 2, fine)
309
+ Token 5: offset +16384 (level 1, medium)
310
+ Token 6: offset +20480 (level 2, fine)
311
+ Token 7: offset +24576 (level 2, fine)
312
+
313
+ Args:
314
+ code_list: List of audio codes (already offset-adjusted)
315
+
316
+ Returns:
317
+ numpy array: Decoded audio waveform
318
+ """
319
+ if not code_list or len(code_list) % 7 != 0:
320
+ raise ValueError("Code list must be non-empty and divisible by 7")
321
+
322
+ # Redistribute codes into SNAC's 3-level hierarchy
323
+ layer_1 = [] # Coarse: 1 token per frame
324
+ layer_2 = [] # Medium: 2 tokens per frame
325
+ layer_3 = [] # Fine: 4 tokens per frame
326
+
327
+ for i in range(len(code_list) // 7):
328
+ # Extract and adjust each token based on its offset
329
+ layer_1.append(code_list[7*i])
330
+ layer_2.append(code_list[7*i + 1] - 4096)
331
+ layer_3.append(code_list[7*i + 2] - (2*4096))
332
+ layer_3.append(code_list[7*i + 3] - (3*4096))
333
+ layer_2.append(code_list[7*i + 4] - (4*4096))
334
+ layer_3.append(code_list[7*i + 5] - (5*4096))
335
+ layer_3.append(code_list[7*i + 6] - (6*4096))
336
+
337
+ # Create hierarchical code tensors for SNAC (on same device as SNAC model)
338
+ snac_device = next(snac_model.parameters()).device
339
+ codes = [
340
+ torch.tensor(layer_1, device=snac_device).unsqueeze(0),
341
+ torch.tensor(layer_2, device=snac_device).unsqueeze(0),
342
+ torch.tensor(layer_3, device=snac_device).unsqueeze(0)
343
+ ]
344
+
345
+ # Decode to audio waveform
346
+ with torch.no_grad():
347
+ audio_hat = snac_model.decode(codes)
348
+
349
+ return audio_hat.squeeze().clamp(-1, 1).cpu().numpy()
350
+
351
+
352
+ # Example usage
353
+ text = "आज का मौसम बहुत अच्छा है। Let's go for a walk!"
354
+ audio = generate_speech(text, temperature=0.5, top_p=0.9)
355
+ sf.write("output.wav", audio, 24000)
356
+ print(f"Generated {len(audio) / 24000:.2f} seconds of audio")
357
+ ```
358
+ ---
359
+
360
+ ## 🌍 Language Support
361
+
362
+ | Language | Status | Notes |
363
+ |----------|--------|-------|
364
+ | **English** | ✅ Supported | Clear, neutral international pronunciation |
365
+ | **Hindi** | ✅ Supported | Natural, fluent, and expressive |
366
+ | **Code‑Mixed (Hinglish)** | ✅ Supported | Seamless language switching |
367
+
368
+ ---
369
+
370
+ ## 🧩 Use Cases
371
+
372
+ Vaakya‑Open is well‑suited for:
373
+
374
+ | Application | Description |
375
+ |-------------|-------------|
376
+ | 📚 **Audiobooks & Narration** | Long‑form content with consistent voice |
377
+ | 🎥 **Video Voice‑Overs** | Professional dubbing and narration |
378
+ | 🏥 **Healthcare** | Patient reminders, medication instructions |
379
+ | 📞 **Voice AI Agents** | IVR systems, conversational assistants |
380
+ | 🧑‍🏫 **E‑learning** | Educational content and tutorials |
381
+ | ♿ **Accessibility** | Screen readers for visually impaired users |
382
+
383
+ ---
384
+
385
+ ## 🏗️ Training Summary
386
+
387
+ | Attribute | Value |
388
+ |-----------|-------|
389
+ | **Speaker Count** | 1 (professional voice‑over artist) |
390
+ | **Recording Quality** | Studio‑grade, 192kHz original capture |
391
+ | **Output Quality** | 24kHz (downsampled for efficiency) |
392
+ | **Data Characteristics** | Conversational, narrative, informational |
393
+ | **Training Method** | LoRA fine‑tuning on Orpheus base |
394
+
395
+ ---
396
+
397
+ ### Training Data Sources
398
+
399
+ This model was trained using high‑quality speech data from publicly available academic datasets developed by premier Indian research institutions:
400
+
401
+ | Dataset | Institution | Description | License |
402
+ |---------|-------------|-------------|---------|
403
+ | **[IndicTTS](https://www.iitm.ac.in/donlab/indictts/database)** | IIT Madras | Hindi and Indian English speech corpus for TTS | Custom (see below) |
404
+ | **[SYSPIN TTS Corpus](https://spiredatasets.iisc.ac.in/syspinCorpus)** | IISc Bangalore / SPIRE Lab | 900+ hours of studio-recorded TTS data in 9 Indian languages | [CC‑BY‑4.0](https://creativecommons.org/licenses/by/4.0/) |
405
+ | **[SPICOR TTS Corpus](https://spiredatasets.iisc.ac.in/spicortts10)** | IISc Bangalore / SPIRE Lab | 97+ hour domain-rich Indian English TTS corpus | [CC‑BY‑4.0](https://creativecommons.org/licenses/by/4.0/) |
406
+
407
+ We gratefully acknowledge these institutions for making their datasets available for research and development in Indian language speech synthesis.
408
+
409
+ ---
410
+
411
+ ## 📊 Performance
412
+
413
+ | Metric | Value |
414
+ |--------|-------|
415
+ | **Latency (A100‑80GB)** | ~120ms |
416
+ | **Latency (RTX 4090)** | ~200ms |
417
+ | **Real‑time Factor** | < 0.1x* |
418
+ | **Output Sample Rate** | 24kHz |
419
+
420
+ > **Note:** Benchmarks measured with 4-bit quantization (`load_in_4bit=True`), batch size 1, average text length ~20-30 tokens, `max_new_tokens=512`. Performance varies significantly based on hardware, precision (FP16/FP32/4-bit), text length, and generation parameters. FP16 inference may be 2-3x slower than 4-bit but provides better quality.
421
+
422
+ ---
423
+
424
+ ## ⚠️ Limitations
425
+
426
+ | Limitation | Description |
427
+ |------------|-------------|
428
+ | **Single Speaker** | No speaker switching or voice selection |
429
+ | **Language Support** | English & Hindi only |
430
+ | **Emotion Control** | Emotion tokens not yet exposed |
431
+ | **Hardware** | GPU recommended for real‑time inference |
432
+
433
+ ---
434
+
435
+ ## 🛣️ Roadmap
436
+
437
+ We are actively working on:
438
+
439
+ - 🎭 Emotion & prosody control tokens
440
+ - 🌏 Additional Indian languages (Tamil, Telugu, Bengali, Marathi)
441
+ - 🎙️ Multiple voice variants (male, regional accents)
442
+ - ⚙️ CPU‑optimized inference paths
443
+ - 📡 Streaming inference support
444
+
445
+ ---
446
+
447
+ ## 📜 License
448
+
449
+ This project is released under the **Apache 2.0 License**.
450
+
451
+ ### Training Data Licenses
452
+
453
+ The training data used in this model is subject to the following licenses:
454
+
455
+ **IIT Madras IndicTTS Dataset:**
456
+ > COPYRIGHT 2016 TTS Consortium, TDIL, Meity represented by Hema A Murthy & S Umesh, DEPARTMENT OF Computer Science and Engineering and Electrical Engineering, IIT Madras. ALL RIGHTS RESERVED.
457
+
458
+ The IndicTTS dataset is provided under a permissive license that allows derivative works and free distribution. See the [full license](https://www.iitm.ac.in/donlab/indictts/database) for details.
459
+
460
+ **IISc Bangalore SYSPIN & SPICOR Datasets:**
461
+
462
+ The SYSPIN and SPICOR TTS corpora are released under the [Creative Commons Attribution 4.0 International License (CC‑BY‑4.0)](https://creativecommons.org/licenses/by/4.0/), which permits sharing and adaptation for any purpose, including commercial use, with appropriate attribution.
463
+
464
+ ---
465
+
466
+ ## 🙏 Acknowledgments
467
+
468
+ ### Training Data
469
+
470
+ We gratefully acknowledge the following institutions and projects for providing high‑quality speech datasets that made this model possible:
471
+
472
+ **Indian Institute of Technology Madras (IIT Madras)**
473
+ - **[IndicTTS Project](https://www.iitm.ac.in/donlab/indictts/database)** — A comprehensive speech corpus for Indian languages developed by Prof. Hema A Murthy, Prof. S Umesh, and the TTS Consortium
474
+ - Funded by: Technology Development for Indian Languages (TDIL), Ministry of Electronics and Information Technology (MeitY), Government of India
475
+
476
+ **Indian Institute of Science, Bangalore (IISc) — SPIRE Lab**
477
+ - **[SYSPIN Project](https://syspin.iisc.ac.in/)** (SYnthesizing SPeech in INdian languages) — 900+ hours of studio-recorded TTS data in 9 Indian languages, led by Prof. Prasanta Kumar Ghosh
478
+ - Funded by: German Development Cooperation "FAIR Forward — AI for All" (GIZ Germany) and Bhashini AI Solutions Private Limited
479
+ - **[SPICOR Project](https://spiredatasets.iisc.ac.in/spicortts10)** — 97+ hours of domain-rich Indian English TTS corpus
480
+ - Dataset URL: [https://spiredatasets.iisc.ac.in/](https://spiredatasets.iisc.ac.in/)
481
+
482
+ ### Architecture & Tools
483
+
484
+ - **[Canopy Labs](https://canopylabs.ai)** — For pioneering the Orpheus TTS architecture and open‑sourcing their work
485
+ - **[Unsloth](https://github.com/unslothai/unsloth)** — For training optimizations and fine‑tuning tools
486
+ - **[SNAC](https://github.com/hubertsiuzdak/snac)** — Hubert Siuzdak for the neural audio codec
487
+
488
+ ### Voice
489
+
490
+ - **Voice Artist** — For providing high‑quality studio recordings
491
+
492
+ ---
493
+
494
+ ## 🛡️ Safety, Ethics & Responsible Use
495
+
496
+ ### Intended Use
497
+
498
+ Vaakya‑Open is designed for **legitimate applications** including:
499
+
500
+ - 📚 Audiobook narration and content creation
501
+ - 🎥 Video voice‑overs and dubbing
502
+ - 🏥 Healthcare communication (patient reminders, medication instructions)
503
+ - 📞 Voice assistants and IVR systems
504
+ - 🧑‍🏫 Educational content and e‑learning
505
+ - ♿ Accessibility tools for visually impaired users
506
+ - 🔬 Academic research in speech synthesis
507
+
508
+ ### Prohibited Uses
509
+
510
+ **Do not use this model for:**
511
+
512
+ - ❌ **Impersonation** — Creating audio that falsely represents a real person without their explicit consent
513
+ - ❌ **Fraud & Scams** — Generating deceptive audio for phishing, vishing, or financial fraud
514
+ - ❌ **Misinformation** — Producing fake news, propaganda, or misleading content
515
+ - ❌ **Deepfakes** — Creating non‑consensual synthetic media intended to deceive
516
+ - ❌ **Harassment** — Generating content to bully, threaten, or demean individuals
517
+ - ❌ **Illegal Activities** — Any use that violates applicable laws or regulations
518
+
519
+ ### Transparency Recommendation
520
+
521
+ If you use Vaakya‑Open to generate speech for public‑facing applications, we **strongly recommend** disclosing to end users that they are listening to AI‑generated content. Transparency builds trust and helps prevent potential misuse.
522
+
523
+ ### Legal Compliance
524
+
525
+ Users are responsible for ensuring their use of this model complies with:
526
+
527
+ - All applicable local, national, and international laws
528
+ - Industry‑specific regulations (healthcare, finance, telecommunications, etc.)
529
+ - Platform terms of service where the generated audio is distributed
530
+ - Intellectual property and privacy rights of third parties
531
+
532
+ ### Disclaimer
533
+
534
+ > **THE DEVELOPERS OF VAAKYA‑OPEN ASSUME NO LIABILITY FOR ANY MISUSE OF THIS MODEL.**
535
+ >
536
+ > This model is provided "as is" without warranty of any kind. Voxaura Labs and its contributors are not responsible for any direct, indirect, incidental, or consequential damages arising from the use or misuse of this model or its outputs.
537
+ >
538
+ > By using this model, you agree to follow all applicable laws, respect the rights and privacy of others, and uphold ethical standards in AI development and deployment.
539
+
540
+ ### Ethical AI Commitment
541
+
542
+ We at **Voxaura Labs** are committed to the responsible development and use of AI technologies. We believe that:
543
+
544
+ 1. **AI should augment human capabilities**, not deceive or harm
545
+ 2. **Transparency** is essential in AI‑generated content
546
+ 3. **Privacy and consent** must be respected in voice applications
547
+ 4. **Accessibility** should be a core consideration in voice technology
548
+
549
+ We encourage the community to use Vaakya‑Open as a force for good — to improve accessibility, enhance communication, and create meaningful experiences while respecting the dignity and rights of all individuals.
550
+
551
+ ### Reporting Misuse
552
+
553
+ If you become aware of any misuse of this model, please report it to us at **[hello@voxygen.ai](mailto:hello@voxygen.ai)**. We take reports of misuse seriously and will take appropriate action where possible.
554
+
555
+ ---
556
+
557
+ ## 💬 About Voxaura Labs & Voxygen.ai
558
+
559
+ **Voxaura Labs** builds advanced Voice AI systems focused on realism, scalability, and multilingual accessibility — bridging the gap between human expression and artificial speech.
560
+
561
+ **Voxygen** is the product brand of Voxaura Labs, dedicated to making advanced Voice AI technologies accessible through practical applications and tools for creators, developers, and enterprises worldwide.
562
+
563
+ 🌐 [Website](https://www.voxygen.ai) · 📧 [Contact](mailto:hello@voxygen.ai)
564
+
565
+ ---
566
+
567
+ ## 📝 Citation
568
+
569
+ ```bibtex
570
+ @misc{vaakya2026,
571
+ title={Vaakya-Open: Text-to-Speech for Hindi and English},
572
+ author={Voxaura Labs},
573
+ year={2026},
574
+ publisher={HuggingFace},
575
+ url={https://huggingface.co/voxaura-labs/vaakya-open}
576
+ }
577
+ ```
578
+
579
+ If you use this model, please also consider citing the training data sources:
580
+
581
+ ```bibtex
582
+ @misc{indictts2016,
583
+ title={IndicTTS: Text-to-Speech for Indian Languages},
584
+ author={Murthy, Hema A and Umesh, S and {TTS Consortium}},
585
+ year={2016},
586
+ institution={Indian Institute of Technology Madras},
587
+ note={TDIL, MeitY},
588
+ url={https://www.iitm.ac.in/donlab/indictts/database}
589
+ }
590
+
591
+ @misc{syspin2025,
592
+ title={SYSPIN_S1.0 Corpus - A TTS Corpus of 900+ hours in nine Indian Languages},
593
+ author={Abhayjeet et al.},
594
+ year={2025},
595
+ institution={Indian Institute of Science, Bengaluru},
596
+ url={https://spiredatasets.iisc.ac.in/syspinCorpus}
597
+ }
598
+
599
+ @misc{spicor2025,
600
+ title={SPICOR TTS_1.0 Corpus - A 97+ hour domain-rich Indian English TTS Corpus},
601
+ author={Abhayjeet et al.},
602
+ year={2025},
603
+ institution={Indian Institute of Science, Bengaluru},
604
+ url={https://spiredatasets.iisc.ac.in/spicortts10}
605
+ }
606
+ ```
607
+
608
+ ---
609
+
610
+ <div align="center">
611
+
612
+ **Made in India 🇮🇳**
613
+
614
+ *If you use Vaakya‑Open in your work, we'd love to hear from you!*
615
+
616
+ </div>
chat_template.jinja ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}
2
+ {%- if custom_tools is defined %}
3
+ {%- set tools = custom_tools %}
4
+ {%- endif %}
5
+ {%- if not tools_in_user_message is defined %}
6
+ {%- set tools_in_user_message = true %}
7
+ {%- endif %}
8
+ {%- if not date_string is defined %}
9
+ {%- if strftime_now is defined %}
10
+ {%- set date_string = strftime_now("%d %b %Y") %}
11
+ {%- else %}
12
+ {%- set date_string = "26 Jul 2024" %}
13
+ {%- endif %}
14
+ {%- endif %}
15
+ {%- if not tools is defined %}
16
+ {%- set tools = none %}
17
+ {%- endif %}
18
+
19
+ {#- This block extracts the system message, so we can slot it into the right place. #}
20
+ {%- if messages[0]['role'] == 'system' %}
21
+ {%- set system_message = messages[0]['content']|trim %}
22
+ {%- set messages = messages[1:] %}
23
+ {%- else %}
24
+ {%- set system_message = "" %}
25
+ {%- endif %}
26
+
27
+ {#- System message #}
28
+ {{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
29
+ {%- if tools is not none %}
30
+ {{- "Environment: ipython\n" }}
31
+ {%- endif %}
32
+ {{- "Cutting Knowledge Date: December 2023\n" }}
33
+ {{- "Today Date: " + date_string + "\n\n" }}
34
+ {%- if tools is not none and not tools_in_user_message %}
35
+ {{- "You have access to the following functions. To call a function, please respond with JSON for a function call." }}
36
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
37
+ {{- "Do not use variables.\n\n" }}
38
+ {%- for t in tools %}
39
+ {{- t | tojson(indent=4) }}
40
+ {{- "\n\n" }}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- system_message }}
44
+ {{- "<|eot_id|>" }}
45
+
46
+ {#- Custom tools are passed in a user message with some extra guidance #}
47
+ {%- if tools_in_user_message and not tools is none %}
48
+ {#- Extract the first user message so we can plug it in here #}
49
+ {%- if messages | length != 0 %}
50
+ {%- set first_user_message = messages[0]['content']|trim %}
51
+ {%- set messages = messages[1:] %}
52
+ {%- else %}
53
+ {{- raise_exception("Cannot put tools in the first user message when there's no first user message!") }}
54
+ {%- endif %}
55
+ {{- '<|start_header_id|>user<|end_header_id|>\n\n' -}}
56
+ {{- "Given the following functions, please respond with a JSON for a function call " }}
57
+ {{- "with its proper arguments that best answers the given prompt.\n\n" }}
58
+ {{- 'Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}.' }}
59
+ {{- "Do not use variables.\n\n" }}
60
+ {%- for t in tools %}
61
+ {{- t | tojson(indent=4) }}
62
+ {{- "\n\n" }}
63
+ {%- endfor %}
64
+ {{- first_user_message + "<|eot_id|>"}}
65
+ {%- endif %}
66
+
67
+ {%- for message in messages %}
68
+ {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
69
+ {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
70
+ {%- elif 'tool_calls' in message %}
71
+ {%- if not message.tool_calls|length == 1 %}
72
+ {{- raise_exception("This model only supports single tool-calls at once!") }}
73
+ {%- endif %}
74
+ {%- set tool_call = message.tool_calls[0].function %}
75
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' -}}
76
+ {{- '{"name": "' + tool_call.name + '", ' }}
77
+ {{- '"parameters": ' }}
78
+ {{- tool_call.arguments | tojson }}
79
+ {{- "}" }}
80
+ {{- "<|eot_id|>" }}
81
+ {%- elif message.role == "tool" or message.role == "ipython" %}
82
+ {{- "<|start_header_id|>ipython<|end_header_id|>\n\n" }}
83
+ {%- if message.content is mapping or message.content is iterable %}
84
+ {{- message.content | tojson }}
85
+ {%- else %}
86
+ {{- message.content }}
87
+ {%- endif %}
88
+ {{- "<|eot_id|>" }}
89
+ {%- endif %}
90
+ {%- endfor %}
91
+ {%- if add_generation_prompt %}
92
+ {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
93
+ {%- endif %}
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 128000,
8
+ "eos_token_id": 128009,
9
+ "head_dim": 128,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 3072,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 8192,
14
+ "max_position_embeddings": 131072,
15
+ "mlp_bias": false,
16
+ "model_type": "llama",
17
+ "num_attention_heads": 24,
18
+ "num_hidden_layers": 28,
19
+ "num_key_value_heads": 8,
20
+ "pad_token_id": 128004,
21
+ "pretraining_tp": 1,
22
+ "rms_norm_eps": 1e-05,
23
+ "rope_scaling": {
24
+ "factor": 32.0,
25
+ "high_freq_factor": 4.0,
26
+ "low_freq_factor": 1.0,
27
+ "original_max_position_embeddings": 8192,
28
+ "rope_type": "llama3"
29
+ },
30
+ "rope_theta": 500000.0,
31
+ "tie_word_embeddings": true,
32
+ "torch_dtype": "bfloat16",
33
+ "transformers_version": "4.55.4",
34
+ "unsloth_fixed": true,
35
+ "unsloth_version": "2026.1.3",
36
+ "use_cache": true,
37
+ "vocab_size": 156940
38
+ }
final_test_1_english.wav ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9d865d7573028f5a8808e76d87de20a5f72b5c8d9950f11ac71182ceb54d4d7
3
+ size 213036
final_test_2_hindi.wav ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73fc7ed45720c0a7e2dddd408e739065681cf745120bcbb4dcd034393a06b0f1
3
+ size 172076
generation_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 128000,
4
+ "do_sample": true,
5
+ "eos_token_id": 128009,
6
+ "max_length": 131072,
7
+ "pad_token_id": 128004,
8
+ "temperature": 0.6,
9
+ "top_p": 0.9,
10
+ "transformers_version": "4.55.4"
11
+ }
model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c54aee36a2479362c44d45c6e3770dc4b06489b4b0f7936f159bd78fb201f526
3
+ size 4991037968
model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:553febe434875a2fcb57cedec5e238e918ca2410c4a2a056c8cffaf6747e6829
3
+ size 1610725592
model.safetensors.index.json ADDED
@@ -0,0 +1,262 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_parameters": 3300867072,
4
+ "total_size": 6601734144
5
+ },
6
+ "weight_map": {
7
+ "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
15
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
16
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
17
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
18
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
19
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
20
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
21
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
22
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
23
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
24
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
25
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
26
+ "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors",
27
+ "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
28
+ "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
29
+ "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
30
+ "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
31
+ "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
32
+ "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
33
+ "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
34
+ "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
35
+ "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors",
36
+ "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
37
+ "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
38
+ "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
39
+ "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
40
+ "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
41
+ "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
42
+ "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
43
+ "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
44
+ "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors",
45
+ "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
46
+ "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
47
+ "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
48
+ "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
49
+ "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
50
+ "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
51
+ "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
52
+ "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
53
+ "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors",
54
+ "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
55
+ "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
56
+ "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
57
+ "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
58
+ "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
59
+ "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
60
+ "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
61
+ "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
62
+ "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors",
63
+ "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
64
+ "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
65
+ "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
66
+ "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
67
+ "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
68
+ "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
69
+ "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
70
+ "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
71
+ "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors",
72
+ "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
73
+ "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
74
+ "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
75
+ "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
76
+ "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
77
+ "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
78
+ "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
79
+ "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
80
+ "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors",
81
+ "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
82
+ "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
83
+ "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
84
+ "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
85
+ "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
86
+ "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
87
+ "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
88
+ "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
89
+ "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors",
90
+ "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
91
+ "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
92
+ "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
93
+ "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
94
+ "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
95
+ "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
96
+ "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
97
+ "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
98
+ "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors",
99
+ "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
100
+ "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
101
+ "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
102
+ "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
103
+ "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
104
+ "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
105
+ "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
106
+ "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
107
+ "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors",
108
+ "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
109
+ "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
110
+ "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
111
+ "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
112
+ "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
113
+ "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
114
+ "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
115
+ "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
116
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors",
117
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
118
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
119
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
120
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
121
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
122
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
123
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
124
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
125
+ "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors",
126
+ "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
127
+ "model.layers.20.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
128
+ "model.layers.20.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
129
+ "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
130
+ "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
131
+ "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
132
+ "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
133
+ "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
134
+ "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors",
135
+ "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
136
+ "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
137
+ "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
138
+ "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
139
+ "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
140
+ "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
141
+ "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
142
+ "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
143
+ "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors",
144
+ "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
145
+ "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
146
+ "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
147
+ "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
148
+ "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
149
+ "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
150
+ "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
151
+ "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
152
+ "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors",
153
+ "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
154
+ "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
155
+ "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
156
+ "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
157
+ "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
158
+ "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
159
+ "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
160
+ "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
161
+ "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors",
162
+ "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
163
+ "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
164
+ "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
165
+ "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
166
+ "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
167
+ "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
168
+ "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
169
+ "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
170
+ "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors",
171
+ "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
172
+ "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
173
+ "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
174
+ "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
175
+ "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
176
+ "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
177
+ "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
178
+ "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
179
+ "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors",
180
+ "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
181
+ "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
182
+ "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
183
+ "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
184
+ "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
185
+ "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
186
+ "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
187
+ "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
188
+ "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors",
189
+ "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
190
+ "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
191
+ "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
192
+ "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
193
+ "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
194
+ "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
195
+ "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
196
+ "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
197
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors",
198
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
199
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
200
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
201
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
202
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
203
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
204
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
205
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
206
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors",
207
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
208
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
209
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
210
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
211
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
212
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
213
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
214
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
215
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors",
216
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
217
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
218
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
219
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
220
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
221
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
222
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
223
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
224
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors",
225
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
226
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
227
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
228
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
229
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
230
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
231
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
232
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
233
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors",
234
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
235
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
236
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
237
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
238
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
239
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
240
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
241
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
242
+ "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors",
243
+ "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
244
+ "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
245
+ "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
246
+ "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
247
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
248
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
249
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
250
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
251
+ "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors",
252
+ "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
253
+ "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
254
+ "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
255
+ "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
256
+ "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
257
+ "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
258
+ "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
259
+ "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
260
+ "model.norm.weight": "model-00002-of-00002.safetensors"
261
+ }
262
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|audio|>"
4
+ ],
5
+ "bos_token": {
6
+ "content": "<|begin_of_text|>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "eos_token": {
13
+ "content": "<|eot_id|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "pad_token": {
20
+ "content": "<|finetune_right_pad_id|>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ }
26
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc3fecb199b4170636dbfab986d25f628157268d37b861f9cadaca60b1353bce
3
+ size 22849547
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff
 
weather.wav ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2580ce07da9778b9cca64684da75186dbfa76156c8ca4d39d4b669c4db59cbcb
3
+ size 192556