This model has been pushed to the Hub using the PytorchModelHubMixin integration:

Installation

First install from here:

git clone -b add_hf https://github.com/NielsRogge/csm.git
cd csm
pip install -r requirements.txt

Usage

import torchaudio
from generator import load_csm_1b

generator = load_csm_1b(device="cuda")

audio = generator.generate(
    text="Hello from Sesame.",
    speaker=0,
    context=[],
    max_audio_length_ms=10_000,
)

torchaudio.save("audio.wav", audio.unsqueeze(0).cpu(), generator.sample_rate)
Downloads last month
32
Safetensors
Model size
1.55B params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.