Skip to main content

Fat Llama Logo

Fat Llama build - status PyPI PyPI - Downloads

fat_llama is a Python package for upscaling audio files to FLAC or WAV formats using advanced audio processing techniques. It utilizes CUDA-accelerated calculations to enhance audio quality by upsampling and adding missing frequencies through FFT (Fast Fourier Transform), resulting in richer and more detailed audio.

Features

  • Upscale MP3 files to high-quality FLAC format.
  • Band-limited (FFT-domain) interpolation and iterative soft thresholding (IST) for enhanced audio processing, with no bandwidth extension beyond the original recording's frequency ceiling.
  • LMS adaptive filtering, auto-scaling amplitude adjustment, and normalization.
  • Supports GPU-accelerated processing with CuPy.
  • Claude is used for coding and maintainance of this package, but not for generating audio.

Requirements

  • CUDA capable GPU

(Note: For cpu verison please look at https://pypi.org/project/fat-llama-fftw/)

Installation

Install via pip:

pip install fat-llama

Note: This version works with CUDA 13 (tested against 13.3).

Further need CUDA & CuPy properly installed: https://docs.cupy.dev/en/stable/install.html

Also, requires ffmpeg: https://support.audacityteam.org/basics/installing-ffmpeg

Note to install on older versions of CUDA and CuPy. You will need to download specific versions and install locally.

To install locally:

git clone <target_url>
cd fat_llama
pip install .

Usage

Example Usage

You can run the example provided in example.py:

from fat_llama.audio_fattener.feed import upscale

# Example call to the method
upscale(
    input_file_path='input_test.mp3',
    output_file_path='output_test.flac',
    source_format='mp3',
    target_format='flac',
    max_iterations=300,
    threshold_value=0.6,
    target_bitrate_kbps=1400,
    toggle_normalize=True,
    toggle_autoscale=True,
    toggle_adaptive_filter=True
)

Function Parameters

  • input_file_path (str): Path to the input audio file. Mandatory.
  • output_file_path (str): Path to the output processed audio file. Mandatory.
  • source_format (str): Format of the input audio file (e.g., 'mp3', 'wav', 'ogg', 'flac').
  • target_format (str): Format of the output audio file (e.g., 'flac', 'wav'). Default is 'flac'.
  • max_iterations (int): Maximum number of iterations for IST. Default is 300.
  • threshold_value (float): Threshold value for IST. Default is 0.6.
  • target_bitrate_kbps (int): Used to derive the upscale factor relative to the source file's own bitrate; must fall within the valid range for target_format (800-1411 kbps for flac, 800-6444 kbps for wav). Default is 1411. Note this is not a promise about the output file's real bitrate — the output is always written as uncompressed PCM at an upsampled sample rate, so its actual bitrate will be substantially higher.
  • toggle_normalize (bool): Whether to normalize the audio. Default True.
  • toggle_autoscale (bool): Whether to autoscale the audio based on the original audio. Default True.
  • toggle_adaptive_filter (bool): Whether to apply LMS adaptive filtering. Default True.

Running the Example

To run the example, execute the following command:

python example.py

This will upscale the MP3 file specified in the example and produce a FLAC file with full processing.

Spectrogram Results

Spectrogram Results

Audio Quality Scores

Generated by the test-fat-llama skill's audio-quality-checker subagent — updated each run, not hand-edited.

Metric Score Notes
Coherence (upscale quality, 0-10) 9 Every check clean and several better than the reference itself - above-22050 Hz energy at -119.1 dB peak (noise floor), clipping 3.70e-5 (reference 1.69e-4), zero dropouts, no NaN/Inf, 99.9th-pct discontinuity well below the reference's, 100ms-window dynamic range 57.1 dB exactly matching the reference (the previous cycle's collapse to 25.4 dB is gone), and no FFT bin persistent across frames (the 98.168 Hz drone is gone). The only below-Nyquist gain is a small content-correlated HF shelf rather than newly reconstructed detail (IST's harmonic term was removed last cycle, so it's now a near-lossless FFT/IFFT round trip) - a safe, transparent upscale rather than a 10.
Spectral deviation vs. reference FLAC (0-10) 9.9 convergence=0.9839, correlation=0.9999 against input_test.flac; both metrics agree, up from 9.3 last cycle - removing IST's harmonic-reconstruction term (which had settled into a static audible drone) resolved the remaining spectral gap with no tradeoff.

How it works

How it Works

Algorithm Explanation

The upscaling process involves several steps:

  1. Reading Audio File: The audio file is read, and the audio samples are extracted along with the sample rate and bitrate.
  2. Calculating Upscale Factor: The upscale factor is derived from target_bitrate_kbps relative to the source file's own bitrate, then clamped so the resulting sample rate never exceeds a realistic consumer playback ceiling (192 kHz) and never goes below the original sample rate — the derivation no longer produces arbitrarily large factors (e.g. 250-300+ kHz output sample rates) that the pipeline's own Nyquist cutoff (step 8) would have made mostly silent anyway.
  3. Upscaling Channels: The audio channels are upscaled using a bandlimited FFT-domain interpolation (zero-padding the spectrum then inverse-transforming) rather than naive sample repetition, so the extra samples don't introduce spectral imaging above the original signal's Nyquist frequency.
  4. Iterative Soft Thresholding (IST): IST is applied to enhance the audio by adding missing frequencies. This process uses FFT to transform the signal into the frequency domain, apply a threshold to keep significant frequencies, and then inverse transform back to the time domain, repeated for max_iterations rounds. A per-iteration synthetic harmonic-reconstruction term was tried across cycles 2-3 (content-derived frequency, peak-scaled amplitude), but was found (cycle 4) to be computed from a single whole-buffer FFT, so for any real multi-second upscaled channel its "content-derived" frequency was actually static across the whole track — a constant, audible, non-source tone that collapsed measured dynamic range in quiet passages. It has been removed rather than revised a fourth time; IST now performs the plain FFT/threshold/IFFT round trip described above with no synthetic tone added on top.
  5. Scaling Amplitude: The amplitude of the upscaled audio is scaled to match the original.
  6. Normalizing Audio: The audio is normalized to the range -1 to 1.
  7. Adaptive Filtering: A block-adaptive LMS filter with a short decorrelation delay refines the normalized signal, adapting its coefficients based on the signal's own short-term predictability. Tap weights update once per block of samples (rather than once per individual sample) using the block-averaged gradient, which is what makes enabling this stage practical at typical upscaled sample counts.
  8. Original-Nyquist Cutoff: An unconditional final FFT-domain lowpass removes any spectral content above the original source file's Nyquist frequency, guaranteeing the upscale never synthesizes or leaves behind content beyond the original recording's real bandwidth — upscaling improves precision and headroom within that bandwidth, it does not extend it.
  9. Writing FLAC File: The processed audio is written to a FLAC file.

Why FFT and IST?

FFT (Fast Fourier Transform) is used to transform the audio signal into the frequency domain. This allows for the identification and manipulation of specific frequency components. By applying a threshold in the frequency domain, we can keep significant frequencies and discard noise and add it to our upscaling data to add detail to upscaling frequencies.

The report titled "Fast Sparse Fourier Transformations for NMR Spectroscopy" by Badruddin Kamal, supervised by Thomas Huber and Alastair Rendall, 2015, provides a comprehensive understanding of sparse representations and their applications in signal processing. IST leverages the concepts from this report to add missing frequencies and enhance the audio quality by making it more detailed and rich. This is particularly useful in upscaling audio where some frequencies might be missing or congested.

Test Audio Source

ericzo - beyond link(https://soundcloud.com/ericzomusic/free-electro-trap-anthem-beyond)

Changelog

The full, current changelog is maintained in CHANGELOG.md — see it for the latest entries going forward. The history below is kept for reference.

[1.4.0] - 2026-09-06

Fixed

  • write_audio() was clipping nearly all output audio (missing normalization before writing PCM).
  • The LMS "adaptive filter" was a silent no-op that burned most of the pipeline's runtime for zero effect.
  • IST's harmonic-reconstruction term was swamped to invisibility at real audio scale.
  • Interpolation used naive sample duplication, causing audible imaging artifacts; replaced with proper band-limited (FFT-based) interpolation — also roughly 1000x faster.
  • Added an unconditional final filter guaranteeing no output content exceeds the original recording's frequency ceiling — upscaling improves precision/headroom within the original bandwidth, it does not extend it.
  • Fixed CI: GitHub's hosted runners have no GPU, so CUDA-dependent tests now skip cleanly there instead of crashing; fixed a stale cupy-cuda12x/cupy-cuda13x version mismatch in the test workflow.

See CHANGELOG.md for full details, including known gaps and measured audio-quality improvements.

[1.1.0] - 2024-08-01

Chanaged

  • Moved adaptive filtering to after normalization and auto-scaling steps.
  • Reduced step size for LMS adaptive filter for improved stability.
  • Ensured all processing uses CuPy for GPU acceleration.
  • Added detailed comments and logging for better traceability.

[1.0.2] - 2024-07-26

Changed

  • Remove logging from requirements to fix pip bug.

[1.0.1] - 2024-07-26

Changed

  • Updated analytics.py analysis and spectorgram results.
  • Updated README.md details.

[1.0.0] - 2024-07-25

Added

  • Added support for reading 'ogg', 'flac', and 'wav' file formats and calculating their bitrates correctly.

Changed

  • Renamed upscale_mp3_to_flac method to upscale to support multiple source formats.
  • Simplified the workflow to focus on 'mp3' to 'flac' conversion with essential steps only.

Removed

  • Dropped support for 'ape' and 'alac' target formats.

[0.1.8] - 2024-07-24

Added

  • Introduced toggle flags for normalization, equalization, amplitude scaling, and gain reduction.
  • Enhanced auto-scaling of amplitude based on the original MP3 file when toggle_scale_amplitude is False.
  • Logging for each step of the processing to provide better traceability and debugging.

Changed

  • Default values for parameters are now set at the function call.
  • Refined the upscaling algorithm to ensure better handling of amplitude and gain.
  • Renamed the flags for consistency (toggle_wiener_filter, toggle_normalize, toggle_equalize, toggle_scale_amplitude, toggle_gain_reduction).

Fixed

  • Fixed issues related to numpy and cupy array conversions.
  • Improved error handling for invalid target bitrate values.
  • Addressed the issue where the amplitude of the produced signal was significantly weaker than the original.

[0.1.7] - 2024-07-22

Added

  • Added methods for MP3 to FLAC conversion with optional processing using CuPy for GPU acceleration.
  • Initial version of upscale_mp3_to_flac method with parameters for iterative soft thresholding (IST), gain reduction, and equalization.

[0.1.0] to [0.1.6] - 2024-07-20

Added

  • Basic functionality for reading MP3 files and writing FLAC files.
  • Initial implementation of the new interpolation algorithm and IST for audio processing.

Metadata

Release files for fat-llama 1.4.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for fat-llama 1.4.2
File Size Uploaded
fat_llama-1.4.2.tar.gz 37.9 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for fat-llama 1.4.2
File Interpreter ABI Platform
fat_llama-1.4.2-py3-none-any.whl Python 3 none any Details

Total release size: 72.3 kB

Release files / fat_llama-1.4.2.tar.gz

Download URL fat_llama-1.4.2.tar.gz
Size 37.9 kB
Tags Source
SHA-256 checksum
How to use checksums
a7c360a08afe65c3f74daafe7571b3e92b48155c49be711a7b181382ce62ddfe
BLAKE2b-256 checksum
How to use checksums
5bae5897568092db744c75749ec6262e2343566def74f41e7d082913900989c0
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.12.14

Release files / fat_llama-1.4.2-py3-none-any.whl

Download URL fat_llama-1.4.2-py3-none-any.whl
Size 34.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
4a9183e6c9ac4a2c26a9561f41ad0ff8674415565b0eaeeeece1f33603be89b8
BLAKE2b-256 checksum
How to use checksums
1bb33e9eb148047a256ebd0f1d742040feda75985bc4801ac81f570910d01a4d
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.12.14

Release history Release notifications | RSS feed

2.0.0

2 release files

1.4.4

2 release files

1.4.3

2 release files

This release

1.4.2 This release

2 release files

1.4.0

2 release files

1.1.0

2 release files

1.0.2

2 release files

1.0.1

2 release files

1.0.0

2 release files

0.1.7

2 release files

0.1.6

2 release files

0.1.5

2 release files

0.1.4

2 release files

0.1.3

2 release files

0.1.2

2 release files

0.1.1

2 release files

0.1.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page