Delete epoch=710-step=258422.ckpt

Removing because it was getting flagged as malware

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.wav filter=lfs diff=lfs merge=lfs -text

examples/generate_emotion_samples.py

@@ -0,0 +1,193 @@
+from __future__ import annotations
+
+import json
+import sys
+from dataclasses import asdict, dataclass
+from pathlib import Path
+
+REPO_ROOT = Path(__file__).resolve().parents[1]
+SRC_DIR = REPO_ROOT / "src"
+if str(SRC_DIR) not in sys.path:
+    sys.path.insert(0, str(SRC_DIR))
+
+from wfloat_tts import SPEAKER_IDS, load_generator, write_wave
+
+
+@dataclass(frozen=True)
+class SampleSpec:
+    voice: str
+    emotion: str
+    intensity: float
+    text: str
+
+
+SAMPLES = (
+    SampleSpec(
+        voice="skilled_hero_man",
+        emotion="joy",
+        intensity=0.7,
+        text="We did it! The bridge is secure and everyone made it across. Take a breath and enjoy this moment with me.",
+    ),
+    SampleSpec(
+        voice="skilled_hero_woman",
+        emotion="neutral",
+        intensity=0.5,
+        text="Stay close. The stones are loose near the edge. We will move carefully and get there together.",
+    ),
+    SampleSpec(
+        voice="fun_hero_man",
+        emotion="surprise",
+        intensity=0.6,
+        text="Wait, that worked? I pulled one lever, the whole wall spun around, and now there is treasure everywhere!",
+    ),
+    SampleSpec(
+        voice="fun_hero_woman",
+        emotion="joy",
+        intensity=0.7,
+        text="Come on, keep up! The crowd is cheering.",
+    ),
+    SampleSpec(
+        voice="strong_hero_man",
+        emotion="anger",
+        intensity=0.80,
+        text="Enough. You had your warning, and you kept pushing innocent people around. Take one more step, and I end this.",
+    ),
+    SampleSpec(
+        voice="strong_hero_woman",
+        emotion="anger",
+        intensity=0.8,
+        text="The chain is holding, and we are not letting this town fall today. Push harder.",
+    ),
+    SampleSpec(
+        voice="mad_scientist_man",
+        emotion="joy",
+        intensity=0.90,
+        text="Look at that reaction. The coils are singing, the chamber is stable, and the entire machine is finally alive. Oh, this is magnificent.",
+    ),
+    SampleSpec(
+        voice="mad_scientist_woman",
+        emotion="surprise",
+        intensity=0.80,
+        text="No, no, that's not possible. The formula should have crystallized, but it adapted instead. Do you realize what that means for the rest of my work?",
+    ),
+    SampleSpec(
+        voice="clever_villain_man",
+        emotion="dismissive",
+        intensity=0.60,
+        text="You came all this way for that little threat? I have heard sharper words from people with far less ambition. Sit down and let the adults finish the game.",
+    ),
+    SampleSpec(
+        voice="clever_villain_woman",
+        emotion="dismissive",
+        intensity=0.74,
+        text="Please. If that was your grand reveal, I expected more style and much better timing. You are already behind, and you still do not know the real plan.",
+    ),
+    SampleSpec(
+        voice="narrator_man",
+        emotion="neutral",
+        intensity=0.5,
+        text="The harbor was quiet again. The storm had passed in the night, leaving scattered crates, broken rope, and one unanswered question.",
+    ),
+    SampleSpec(
+        voice="narrator_woman",
+        emotion="sadness",
+        intensity=0.4,
+        text="When the letter finally arrived, no one rushed to open it. The room had grown too familiar with bad news, and hope had learned to speak softly.",
+    ),
+    SampleSpec(
+        voice="wise_elder_man",
+        emotion="anger",
+        intensity=0.4,
+        text="I have seen this mistake before. Pride speaks loudly in the beginning, but regret is the voice that stays with you. Listen while there is still time to change course.",
+    ),
+    SampleSpec(
+        voice="wise_elder_woman",
+        emotion="joy",
+        intensity=0.5,
+        text="There you are. The answer was never as far away as you feared, and your patience has finally borne fruit. Let yourself be proud of how far you have come.",
+    ),
+    SampleSpec(
+        voice="outgoing_anime_man",
+        emotion="joy",
+        intensity=0.8,
+        text="Yes. This is our chance, and I can feel the whole day lighting up around us. We are going in together, and we are coming out legends.",
+    ),
+    SampleSpec(
+        voice="outgoing_anime_woman",
+        emotion="surprise",
+        intensity=0.5,
+        text="Hold on, you are telling me that tiny mascot can drive the entire ship? That is completely ridiculous, and somehow it makes me want to see it even more.",
+    ),
+    SampleSpec(
+        voice="scary_villain_man",
+        emotion="anger",
+        intensity=0.5,
+        text="You should have stayed hidden. Now I know your face, your voice, and exactly how afraid you are trying not to sound. Run, if it helps you feel alive.",
+    ),
+    SampleSpec(
+        voice="scary_villain_woman",
+        emotion="fear",
+        intensity=0.80,
+        text="Did you hear that? The halls were silent a moment ago, and now something is moving behind the walls. Do not leave me alone with that sound.",
+    ),
+    SampleSpec(
+        voice="news_reporter_man",
+        emotion="neutral",
+        intensity=0.5,
+        text="We are live outside city hall, where officials say the meeting will begin within the hour. Security has tightened, and a large crowd is still gathering behind us.",
+    ),
+    SampleSpec(
+        voice="news_reporter_woman",
+        emotion="neutral",
+        intensity=0.6,
+        text="We are receiving conflicting reports from the scene. One agency says the system is back online, while another says key services are still down. We are working to confirm the facts.",
+    ),
+)
+
+
+def intensity_slug(value: float) -> str:
+    return f"{int(round(value * 100)):03d}"
+
+
+def sample_filename(index: int, sample: SampleSpec) -> str:
+    return f"{index:02d}_{sample.voice}_{sample.emotion}_{intensity_slug(sample.intensity)}.wav"
+
+
+def main() -> None:
+    output_dir = REPO_ROOT / "samples"
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+    generator = load_generator(
+        checkpoint_path=REPO_ROOT / "model.safetensors",
+        config_path=REPO_ROOT / "config.json",
+    )
+
+    manifest: list[dict[str, object]] = []
+
+    for index, sample in enumerate(SAMPLES, start=1):
+        sid = SPEAKER_IDS[sample.voice]
+        audio = generator.generate(
+            text=sample.text,
+            sid=sid,
+            emotion=sample.emotion,
+            intensity=sample.intensity,
+        )
+        filename = sample_filename(index, sample)
+        output_path = output_dir / filename
+        write_wave(output_path, audio.samples, audio.sample_rate)
+        manifest.append(
+            {
+                "filename": filename,
+                "sid": sid,
+                **asdict(sample),
+            }
+        )
+        print(output_path.relative_to(REPO_ROOT))
+
+    manifest_path = output_dir / "manifest.json"
+    manifest_path.write_text(json.dumps(manifest, indent=2) + "\n", encoding="utf-8")
+    print(manifest_path.relative_to(REPO_ROOT))
+
+
+if __name__ == "__main__":
+    main()

samples/01_skilled_hero_man_joy_070.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9dd30979c71a6fd9bfeff98cd0142684a25eef9a544730f4ffd691286a6d9e1
+size 236076

samples/02_skilled_hero_woman_neutral_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d07939b3ced69fc3946e6473ae458843809c7c31c01427c72fe04e089cf7c1b5
+size 204844

samples/03_fun_hero_man_surprise_060.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca4a878cfbf27d78ede4419a8e3ffc34095307d667cd0a4b9956d046f23c8e60
+size 264748

samples/04_fun_hero_woman_joy_070.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f89d36d2ca1d149f2eb6afc26a5e34cb81c87801f811aac5d9fd79bfa9c8882
+size 103980

samples/05_strong_hero_man_anger_080.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:baa04c1e568a213ed430cf11aacb0a71969818a1fdb4a218999aa4a85a708b96
+size 307244

samples/06_strong_hero_woman_anger_080.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3eb50d9f75ea3fb7a0017bfd668d523c3543cffddce127a68a81cf4966f52af9
+size 209452

samples/07_mad_scientist_man_joy_090.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d1069c891e96c8e0cdbf6858c47e391b6319de1a723ce2e486bcb2d5518ed23
+size 415276

samples/08_mad_scientist_woman_surprise_080.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e64626702148e9eb0ed6d8fa8edcb34278521c38af113598b521dd96194d4ec0
+size 372780

samples/09_clever_villain_man_dismissive_060.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac4315e4dc5e19dd5eb93f636985e73f9983e2bb5a443c9240e7805e65897fdd
+size 338988

samples/10_clever_villain_woman_dismissive_074.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:585cbe0fb2e8ec8e500528bec5a46d7401f67d37dd1fc0cfc2a5c5bc6fb54a4c
+size 391724

samples/11_narrator_man_neutral_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:670f9273663ae845cd283ecec295642994406a15f487e45780f6c296b8268afc
+size 348204

samples/12_narrator_woman_sadness_040.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:654757ddaeabf9b4b004e48b626438218cc0d19467815c714ddf7e0c8e1ea45b
+size 346156

samples/13_wise_elder_man_anger_040.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4b48e59481bd6f764c8cace0c09bd9acc7821b8e90587572a0923ba7c1ac609
+size 343596

samples/14_wise_elder_woman_joy_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71d6efe48bafe8ebbf4341f6a6e63b817c0f7a9dd9021a72d1d41dfdf5544e4f
+size 365612

samples/15_outgoing_anime_man_joy_080.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c19313088762b13130ad898b70e90b397449864857cb0fdc398f8075b7f23128
+size 310316

samples/16_outgoing_anime_woman_surprise_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30491594ecd680c21bb01b32cde72fe78c04e926b6ba5e0cd78da821ab75e6ee
+size 354860

samples/17_scary_villain_man_anger_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0b1a7b8d91b509e6ebf167dc483c2f4746695e3611ca0b77acf972697b71a13
+size 382508

samples/18_scary_villain_woman_fear_080.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6be568ede696e7ae877c31d09705eb1ce2a30784b28f0629c573421824194683
+size 321580

samples/19_news_reporter_man_neutral_050.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8fdc2eaa384e09007ba76bfc672caa38c4d413db4897ec47d73a641fd758cf91
+size 353324

samples/20_news_reporter_woman_neutral_060.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd95657ded3cd70926c65537087d72f4bfb1371e4b17a6f0851320f6ea0e2249
+size 413228

samples/manifest.json

@@ -0,0 +1,162 @@
+[
+  {
+    "filename": "01_skilled_hero_man_joy_070.wav",
+    "sid": 0,
+    "voice": "skilled_hero_man",
+    "emotion": "joy",
+    "intensity": 0.7,
+    "text": "We did it! The bridge is secure and everyone made it across. Take a breath and enjoy this moment with me."
+  },
+  {
+    "filename": "02_skilled_hero_woman_neutral_050.wav",
+    "sid": 1,
+    "voice": "skilled_hero_woman",
+    "emotion": "neutral",
+    "intensity": 0.5,
+    "text": "Stay close. The stones are loose near the edge. We will move carefully and get there together."
+  },
+  {
+    "filename": "03_fun_hero_man_surprise_060.wav",
+    "sid": 2,
+    "voice": "fun_hero_man",
+    "emotion": "surprise",
+    "intensity": 0.6,
+    "text": "Wait, that worked? I pulled one lever, the whole wall spun around, and now there is treasure everywhere!"
+  },
+  {
+    "filename": "04_fun_hero_woman_joy_070.wav",
+    "sid": 3,
+    "voice": "fun_hero_woman",
+    "emotion": "joy",
+    "intensity": 0.7,
+    "text": "Come on, keep up! The crowd is cheering."
+  },
+  {
+    "filename": "05_strong_hero_man_anger_080.wav",
+    "sid": 4,
+    "voice": "strong_hero_man",
+    "emotion": "anger",
+    "intensity": 0.8,
+    "text": "Enough. You had your warning, and you kept pushing innocent people around. Take one more step, and I end this."
+  },
+  {
+    "filename": "06_strong_hero_woman_anger_080.wav",
+    "sid": 5,
+    "voice": "strong_hero_woman",
+    "emotion": "anger",
+    "intensity": 0.8,
+    "text": "The chain is holding, and we are not letting this town fall today. Push harder."
+  },
+  {
+    "filename": "07_mad_scientist_man_joy_090.wav",
+    "sid": 6,
+    "voice": "mad_scientist_man",
+    "emotion": "joy",
+    "intensity": 0.9,
+    "text": "Look at that reaction. The coils are singing, the chamber is stable, and the entire machine is finally alive. Oh, this is magnificent."
+  },
+  {
+    "filename": "08_mad_scientist_woman_surprise_080.wav",
+    "sid": 7,
+    "voice": "mad_scientist_woman",
+    "emotion": "surprise",
+    "intensity": 0.8,
+    "text": "No, no, that's not possible. The formula should have crystallized, but it adapted instead. Do you realize what that means for the rest of my work?"
+  },
+  {
+    "filename": "09_clever_villain_man_dismissive_060.wav",
+    "sid": 8,
+    "voice": "clever_villain_man",
+    "emotion": "dismissive",
+    "intensity": 0.6,
+    "text": "You came all this way for that little threat? I have heard sharper words from people with far less ambition. Sit down and let the adults finish the game."
+  },
+  {
+    "filename": "10_clever_villain_woman_dismissive_074.wav",
+    "sid": 9,
+    "voice": "clever_villain_woman",
+    "emotion": "dismissive",
+    "intensity": 0.74,
+    "text": "Please. If that was your grand reveal, I expected more style and much better timing. You are already behind, and you still do not know the real plan."
+  },
+  {
+    "filename": "11_narrator_man_neutral_050.wav",
+    "sid": 10,
+    "voice": "narrator_man",
+    "emotion": "neutral",
+    "intensity": 0.5,
+    "text": "The harbor was quiet again. The storm had passed in the night, leaving scattered crates, broken rope, and one unanswered question."
+  },
+  {
+    "filename": "12_narrator_woman_sadness_040.wav",
+    "sid": 11,
+    "voice": "narrator_woman",
+    "emotion": "sadness",
+    "intensity": 0.4,
+    "text": "When the letter finally arrived, no one rushed to open it. The room had grown too familiar with bad news, and hope had learned to speak softly."
+  },
+  {
+    "filename": "13_wise_elder_man_anger_040.wav",
+    "sid": 12,
+    "voice": "wise_elder_man",
+    "emotion": "anger",
+    "intensity": 0.4,
+    "text": "I have seen this mistake before. Pride speaks loudly in the beginning, but regret is the voice that stays with you. Listen while there is still time to change course."
+  },
+  {
+    "filename": "14_wise_elder_woman_joy_050.wav",
+    "sid": 13,
+    "voice": "wise_elder_woman",
+    "emotion": "joy",
+    "intensity": 0.5,
+    "text": "There you are. The answer was never as far away as you feared, and your patience has finally borne fruit. Let yourself be proud of how far you have come."
+  },
+  {
+    "filename": "15_outgoing_anime_man_joy_080.wav",
+    "sid": 14,
+    "voice": "outgoing_anime_man",
+    "emotion": "joy",
+    "intensity": 0.8,
+    "text": "Yes. This is our chance, and I can feel the whole day lighting up around us. We are going in together, and we are coming out legends."
+  },
+  {
+    "filename": "16_outgoing_anime_woman_surprise_050.wav",
+    "sid": 15,
+    "voice": "outgoing_anime_woman",
+    "emotion": "surprise",
+    "intensity": 0.5,
+    "text": "Hold on, you are telling me that tiny mascot can drive the entire ship? That is completely ridiculous, and somehow it makes me want to see it even more."
+  },
+  {
+    "filename": "17_scary_villain_man_anger_050.wav",
+    "sid": 16,
+    "voice": "scary_villain_man",
+    "emotion": "anger",
+    "intensity": 0.5,
+    "text": "You should have stayed hidden. Now I know your face, your voice, and exactly how afraid you are trying not to sound. Run, if it helps you feel alive."
+  },
+  {
+    "filename": "18_scary_villain_woman_fear_080.wav",
+    "sid": 17,
+    "voice": "scary_villain_woman",
+    "emotion": "fear",
+    "intensity": 0.8,
+    "text": "Did you hear that? The halls were silent a moment ago, and now something is moving behind the walls. Do not leave me alone with that sound."
+  },
+  {
+    "filename": "19_news_reporter_man_neutral_050.wav",
+    "sid": 18,
+    "voice": "news_reporter_man",
+    "emotion": "neutral",
+    "intensity": 0.5,
+    "text": "We are live outside city hall, where officials say the meeting will begin within the hour. Security has tightened, and a large crowd is still gathering behind us."
+  },
+  {
+    "filename": "20_news_reporter_woman_neutral_060.wav",
+    "sid": 19,
+    "voice": "news_reporter_woman",
+    "emotion": "neutral",
+    "intensity": 0.6,
+    "text": "We are receiving conflicting reports from the scene. One agency says the system is back online, while another says key services are still down. We are working to confirm the facts."
+  }
+]