{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "19b4f6ae",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T14:58:52.880444Z",
"start_time": "2024-02-15T14:58:52.876108Z"
}
},
"outputs": [],
"source": [
"import os\n",
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\""
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "1fc7b73d",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T14:58:54.632999Z",
"start_time": "2024-02-15T14:58:52.882406Z"
}
},
"outputs": [],
"source": [
"%matplotlib inline\n",
"from matplotlib import pyplot as plt"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "3276f7d5",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T14:58:57.498614Z",
"start_time": "2024-02-15T14:58:54.634423Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"WILL USE FLASH ATTN: True\n",
"WARNING: feature_extractor_cqt requires the libray 'nnAudio'\n"
]
}
],
"source": [
"import numpy as np\n",
"from suno_utils.audio import Audio\n",
"from suno_utils.gpt.chirp_v2_5 import generate_audio, preload_models, codec_decode, GenerationConfig"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "75e60629",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T14:59:15.113681Z",
"start_time": "2024-02-15T14:58:57.500358Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"model loaded: /app/suno/checkpoints/2024-02-09_15-32-17/last_ckpt_infer.pt\n",
"model loaded: 728.0M params\n"
]
}
],
"source": [
"_ = preload_models(\n",
" # gpt_ckpt_path=\"/app/suno/checkpoints/2024-02-04_06-36-38/last_ckpt_infer.pt\", # lg base\n",
" gpt_ckpt_path=\"/app/suno/checkpoints/2024-02-09_15-32-17/last_ckpt_infer.pt\", #sliding window, rotary\n",
" tokenizer_path=\"/app/suno/models/chirp_v2/tokenizer_60k.json\",\n",
" semantic_ckpt_path=\"/app/suno/models/chirp_v2/mert_25.pt\",\n",
" semantic_centroids_path=\"/app/suno/models/chirp_v2/mert_25_2x4k.npy\",\n",
" codec_ckpt_path=\"/app/suno/models/chirp_v2/dac_2c_25x12.pt\",\n",
" hoot_ckpt_path=\"/app/suno/models/chirp_v2/hoot_ckpt.pt\",\n",
" hoot_tokenizer_path=\"/app/suno/models/chirp_v2/tokenizer.model\",\n",
" fasttext_ckpt_path=\"/app/suno/models/chirp_v2/lid.176.bin\",\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "6442b74c",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T14:59:35.663596Z",
"start_time": "2024-02-15T14:59:15.116213Z"
}
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████| 605/605 [00:09<00:00, 63.06it/s]\n"
]
},
{
"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": [
"cfg = GenerationConfig(\n",
" n_batch=4,\n",
" max_gen_duration_s=20,\n",
")\n",
"out_history_arrays, gen_raw_arrays, gen_audios = generate_audio(cfg, return_raw_arrays=True)\n",
"for e in gen_audios:\n",
" e.play()"
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "ed017ae1",
"metadata": {
"ExecuteTime": {
"end_time": "2024-02-15T16:24:16.674468Z",
"start_time": "2024-02-15T16:22:22.290761Z"
}
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
" 94%|█████████████████████████████████████████████████████████████████████████████████████████████████████▊ | 2928/3105 [00:47<00:02, 61.38it/s]\n"
]
},
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"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": [
"text = \"\"\"\n",
"[Verse 1]\n",
"la la la la la la la one \n",
"la la la la la la la two\n",
"la la la la la la la three\n",
"\n",
"[Verse 1]\n",
"Almost Heaven, West Virginia\n",
"Blue Ridge Mountains, Shenandoah River\n",
"Life is old there, older than the trees\n",
"Younger than the mountains, growing like a breeze\n",
"\n",
"[Chorus]\n",
"Country roads, take me home\n",
"To the place I belong\n",
"West Virginia, mountain mama\n",
"Take me home, country roads\n",
"\n",
"[Verse 2]\n",
"la la la la la la la one \n",
"la la la la la la la two\n",
"la la la la la la la three\n",
"\"\"\"\n",
"\n",
"cfg = GenerationConfig(\n",
" text=text,\n",
" cfg_coef=1.5,\n",
" text_tags=\"bluegrass, female\",\n",
" n_batch=4,\n",
" max_gen_duration_s=120,\n",
")\n",
"for e in generate_audio(cfg):\n",
" e.play()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "69e1ae85",
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"id": "394155b3",
"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.10.13"
},
"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
}