Sound-Broken Nemotron LoRA Adapter

Fine-tuned LoRA adapter for "Does It Sound Broken?" โ€” an appliance fault diagnosis app from the Build Small Hackathon 2026.

What it does

This adapter teaches Nemotron-Nano-4B to produce grounded JSON diagnoses from deterministic audio features + rule-engine candidates. The model never hears raw audio โ€” it reasons over 14 measured acoustic features (spectral centroid, RMS, onset rate, harmonic ratio, etc.) extracted by librosa.

Training

  • Base model: nvidia/NVIDIA-Nemotron-3-Nano-4B-BF16
  • Dataset: DCASE 2025 Task 2 (real industrial machine audio, 7 machine types)
  • Training pairs: 300 (feature descriptions + rule candidates โ†’ JSON responses)
  • Epochs: 2
  • LoRA config: r=16, alpha=32, dropout=0.05, targets: q/k/v/o_proj
  • Loss convergence: 1.85 โ†’ 0.72 (epoch 1, step 100)

Usage with PEFT

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

base = AutoModelForCausalLM.from_pretrained(
    "nvidia/NVIDIA-Nemotron-3-Nano-4B-BF16", torch_dtype="auto"
)
model = PeftModel.from_pretrained(base, "mitvho09/sound-broken-nemotron-lora")
tok = AutoTokenizer.from_pretrained("mitvho09/sound-broken-nemotron-lora")

Context

Part of the "Does It Sound Broken?" app โ€” record 10 seconds of an appliance, get a diagnosis. The full pipeline:

  1. librosa extracts 14 deterministic features (CPU)
  2. Rule engine ranks candidate faults (CPU, transparent)
  3. This LoRA (on Nemotron-4B) produces grounded JSON explanation (GPU)
  4. json_guard validates output grounding (CPU)

The rule engine is the floor; this adapter makes the LLM narration more accurate.

Downloads last month
8
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for mitvho09/sound-broken-nemotron-lora

Space using mitvho09/sound-broken-nemotron-lora 1