Csm 1b

🚀 クイックスタート

このモデルは、PytorchModelHubMixin の統合を使用してハブにプッシュされています。

• ライブラリ: https://github.com/SesameAILabs/csm

📦 インストール

まずはここからインストールします。

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

💻 使用例

基本的な使用法

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)

📄 ライセンス

Apache-2.0