import torch
import soundfile as sf
from qwen_tts import Qwen3TTSModel

# 모델 로드
model = Qwen3TTSModel.from_pretrained(
    "daje/Qwen3-TTS-12Hz-0.6B-Korean",
    device_map="cuda:0",
    dtype=torch.bfloat16,
)

# 사용 가능한 화자 확인
print(model.get_supported_speakers())  # ['korean_tts']

# 음성 생성
text = "안녕하세요, 한국어 TTS 모델입니다."
wavs, sr = model.generate_custom_voice(text=text, speaker="korean_tts")

# 파일로 저장
sf.write("output.wav", wavs[0], sr)