Downloads · 30 days
0
audio-restore
todteera/audio-restore
audio-restore is a audio-to-audio model from todteera. Use it for the audio-to-audio task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as cc-by-nc-4.0.
Repairs speech recordings damaged by clippings. A generative model reconstructs the samples that were destroyed.
Access
Public
Updated Aug 11, 2026
Repo size
196 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt196 MB · 100%
At a glance
- Task
- Audio-to-Audio
- Library
- pytorch
- License
- cc-by-nc-4.0
- Access
- Public
- Created
- Aug 11, 2026
- Updated
- Aug 11, 2026
- SHA
- 310b0c11
Model card
From the Hugging Face model README
audio-restore
Repairs speech recordings damaged by clippings. A generative model reconstructs the samples that were destroyed.
Usage
git clone https://github.com/tdstt22/audio-restore
cd audio-restore && uv sync
python restore.py recording.wav restored.wav
Components
| File | Description |
|---|---|
model.pt | 2D U-Net, 48.9M params — flow-matching velocity predictor over complex STFT. EMA weights plus architecture config |
The model works on spectrograms at 24 kHz mono. It is given the damaged audio and a mask marking which samples the clipping destroyed, so it never has to guess where the damage is. Rather than emitting audio directly, it predicts the direction from noise toward clean speech, and sampling follows that direction over to arrive at the reconstruction.
Limitations
- 24 kHz — higher-rate files are converted in and out, so repaired regions carry nothing above 12 kHz. Audible as dullness on sibilants.
- Speech only — music and general audio are out of distribution.
- Hard clipping only — only focused on hard clippings that require regeneration
- Degradations above ~30% clipped — did not optimize for these situations during training
- Task
- Audio-to-Audio
- Library
- pytorch
- License
- cc-by-nc-4.0
- Languages
- en
- Created
- Aug 11, 2026
- Updated
- Aug 11, 2026