{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "3abd2681",
"metadata": {},
"outputs": [],
"source": [
"import os\n",
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n",
"\n",
"from suno_utils.audio import Audio\n",
"from suno_utils.tasks.mert_v2 import (\n",
" encode as mert_encode,\n",
" preload_models as preload_mert_models\n",
")\n",
"from suno_utils.tasks.dac import (\n",
" encode as codec_encode,\n",
" decode as codec_decode,\n",
" preload_models as preload_codec_models,\n",
")\n",
"\n",
"from suno_utils.tasks.gpt_v2.chirp_v1 import preload_models as preload_chirp_models\n",
"from suno_utils.tasks.gpt_v2.chirp_v1 import generate_audio\n",
"from suno_utils.tasks.gpt_v2.generation import generate"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "f97f7e06",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"model loaded: 1720.9M params, 1.52 loss\n"
]
}
],
"source": [
"preload_chirp_models(\n",
" centroids_filepath=\"/home/georg/notebooks/gpt/data/cluster_centers/2x1k_centroids_mert.npy\",\n",
" #gpt_ckpt_path=\"/mnt/data/georg/models/chirp_v1/xl_combo.pt\",\n",
" gpt_ckpt_path=\"/mnt/data/georg/models/chirp_v1/xl_s5.pt\",\n",
" codec_ckpt_path=\"/mnt/data/georg/tmp/d_codec_25x8.pt\",\n",
" common_genre_tags_path=\"/home/georg/notebooks/gpt/chirp_v1/common_tags.json\",\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "cceb2397",
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"100%|██████████████████████████████████████████████████████████████████▉| 3182/3185 [02:34<00:00, 20.58it/s]\n"
]
}
],
"source": [
"text = \"\"\"\n",
"[Intro]\n",
"Yeah\n",
"\n",
"[Verse]\n",
"I've been tryna call\n",
"I've been on my own for long enough\n",
"Maybe you can show me how to love, maybe\n",
"\n",
"[Verse]\n",
"I'm going through withdrawals\n",
"You don't even have to do too much\n",
"You can turn me on with just a touch, baby\n",
"I look around and\n",
"\n",
"[Pre-chorus]\n",
"Sin City's cold and empty (oh)\n",
"No one's around to judge me (oh)\n",
"I can't see clearly when you're gone\n",
"\n",
"[Chorus]\n",
"I said, ooh, I'm blinded by the lights\n",
"No, I can't sleep until I feel your touch\n",
"I said, ooh, I'm drowning in the night\n",
"Oh, when I'm like this, you're the one I trust\n",
"(Hey, hey, hey)\n",
"\n",
"\n",
"\"\"\"\n",
"out = generate_audio(\n",
" text,\n",
" text_tags=\"[k-pop melodic]\", \n",
" n_batch=2,\n",
" history_audio=None,\n",
" history_text=None,\n",
" max_gen_duration_s=40,\n",
" max_history_duration_s=10.01,\n",
" temp_semantic=0.9,\n",
" temp_coarse=0.8,\n",
" top_k_semantic=250,\n",
" top_k_coarse=500,\n",
" cfg_coef=1.3,\n",
" cfg_coef_tags=0,\n",
" min_eos_p=0.25,\n",
" rep_penality=0.2,\n",
" allow_genre_randomization=True,\n",
" return_raw_array=True,\n",
" silent=False,\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "f1cbb567",
"metadata": {},
"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": [
"#codec_decode(out[1][1][0]).play()\n",
"for o in out[0]:\n",
" o.play()\n",
"# or out[0][0].play()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "49eed3b9",
"metadata": {},
"outputs": [],
"source": []
}
],
"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.8.15"
}
},
"nbformat": 4,
"nbformat_minor": 5
}