Views
No views yet
1import soundfile as sf
2from diffusers import LongCatAudioDiTPipeline
3import torch
4
5pipeline = LongCatAudioDiTPipeline.from_pretrained(
6 "ruixiangma/LongCat-AudioDiT-1B-Diffusers",
7 torch_dtype=torch.bfloat16
8)
9
10pipeline = pipeline.to("cuda")
11
12prompt = "A calm ocean wave ambience with soft wind in the background."
13audio = pipeline(prompt, audio_duration_s=5.0, num_inference_steps=20, guidance_scale=4.0, seed=42).audios[0, 0]
14
15sf.write("output.wav", audio, pipeline.sample_rate)