{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "8fae8f9c",
"metadata": {
"ExecuteTime": {
"end_time": "2023-10-18T00:05:43.932984Z",
"start_time": "2023-10-18T00:05:40.790354Z"
}
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/home/tony/anaconda3/envs/suno_env/lib/python3.10/site-packages/whisper/timing.py:57: NumbaDeprecationWarning: \u001b[1mThe 'nopython' keyword argument was not supplied to the 'numba.jit' decorator. The implicit default value for this argument is currently False, but it will be changed to True in Numba 0.59.0. See https://numba.readthedocs.io/en/stable/reference/deprecation.html#deprecation-of-object-mode-fall-back-behaviour-when-using-jit for details.\u001b[0m\n",
" def backtrace(trace: np.ndarray):\n"
]
}
],
"source": [
"import os\n",
"\n",
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"\"\n",
"\n",
"import numpy as np\n",
"from suno_utils.audio import Audio\n",
"from suno_utils.gpt.chirp_v2 import (\n",
" generate_audio,\n",
" generate_audio_stream,\n",
" preload_models,\n",
" codec_decode,\n",
" GenerationConfig,\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "9108708b",
"metadata": {
"ExecuteTime": {
"end_time": "2023-10-18T00:05:53.875147Z",
"start_time": "2023-10-18T00:05:43.934745Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"model loaded: 2615.5M params\n"
]
}
],
"source": [
"_ = preload_models(\n",
" gpt_ckpt_path=\"/home/victor/data/models/chirp_v2/xxl.pt\",\n",
" tokenizer_path=\"/home/victor/data/models/chirp_v2/tokenizer_60k.json\",\n",
" semantic_ckpt_path=\"/home/victor/data/models/chirp_v2/mert_25.pt\",\n",
" semantic_centroids_path=\"/home/victor/data/models/chirp_v2/mert_25_2x4k.npy\",\n",
" codec_ckpt_path=\"/home/victor/data/models/chirp_v2/dac_2c_25x8.pt\",\n",
" fasttext_ckpt_path=\"/home/victor/data/models/lid.176.bin\",\n",
" load_whisper=False,\n",
" use_gpu=False\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "072adcda",
"metadata": {
"ExecuteTime": {
"end_time": "2023-10-18T00:05:53.878843Z",
"start_time": "2023-10-18T00:05:53.876717Z"
}
},
"outputs": [],
"source": [
"text = \"\"\"\n",
"[Verse 1]\n",
"一盏离愁孤灯伫立在窗口\n",
"我在门后假装你人还没走\n",
"旧地如重游月圆更寂寞\n",
"夜半清醒的烛火不忍苛责我\n",
"\n",
"一壶漂泊浪迹天涯难入喉\n",
"你走之后酒暖回忆思念瘦\n",
"水向东流时间怎么偷\n",
"花开就一次成熟我却错过\n",
"\n",
"[Chorus]\n",
"谁在用琵琶弹奏一曲东风破\n",
"岁月在墙上剥落看见小时候\n",
"犹记得那年我们都还很年幼\n",
"而如今琴声幽幽我的等候你没听过\n",
"\n",
"谁在用琵琶弹奏一曲东风破\n",
"枫叶将故事染色结局我看透\n",
"篱笆外的古道我牵着你走过\n",
"荒烟蔓草的年头就连分手都很沉默\n",
"\"\"\""
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "c12aceec",
"metadata": {
"ExecuteTime": {
"end_time": "2023-10-18T00:07:55.973396Z",
"start_time": "2023-10-18T00:05:53.880478Z"
}
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████| 585/585 [01:54<00:00, 5.13it/s]\n"
]
}
],
"source": [
"out_audio = generate_audio(\n",
" GenerationConfig(\n",
" text=text,\n",
" text_tags=\"pop\",\n",
" n_batch=2,\n",
" max_gen_duration_s=20,\n",
" # stream=False,\n",
" )\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "73dd0216",
"metadata": {
"ExecuteTime": {
"end_time": "2023-10-18T00:07:56.656150Z",
"start_time": "2023-10-18T00:07:55.975360Z"
}
},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"for e in out_audio:\n",
" e.play()"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.12"
},
"toc": {
"base_numbering": 1,
"nav_menu": {},
"number_sections": true,
"sideBar": true,
"skip_h1_title": false,
"title_cell": "Table of Contents",
"title_sidebar": "Contents",
"toc_cell": false,
"toc_position": {},
"toc_section_display": true,
"toc_window_display": false
}
},
"nbformat": 4,
"nbformat_minor": 5
}