{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "eb0fbedb",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:14.007379Z",
"start_time": "2024-04-10T00:42:14.005878Z"
}
},
"outputs": [],
"source": [
"import os\n",
"\n",
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"3\""
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "b794cca2",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:33.492428Z",
"start_time": "2024-04-10T00:42:33.490469Z"
}
},
"outputs": [],
"source": [
"from suno_utils.tasks.hoot import encode, load_model_list\n",
"from suno_utils.audio import Audio"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "01ba00f3",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:20.140589Z",
"start_time": "2024-04-10T00:42:15.750010Z"
}
},
"outputs": [],
"source": [
"model_list = load_model_list(\n",
" # checkpoint_filepath=os.path.join(\"/home/tony/Data/checkpoints/hoot/2023-11-17_15-08-57\", \"28k_ckpt.pt\"),\n",
" checkpoint_filepath=os.path.join(\n",
" \"/home/tony/Data/checkpoints/hoot/2024-01-12_22-51-29\", \"170k_ckpt.pt\"\n",
" ),\n",
" tokenizer_filepath=\"/home/tony/Work/tony/hoot/tokenizers/full/tokenizer_spe_bpe_v20480/tokenizer.model\",\n",
" # n_gpus=0,\n",
" )"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "a560be7f",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:36.499276Z",
"start_time": "2024-04-10T00:42:35.450360Z"
}
},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"input_audio = Audio.from_file(\"../audios/test_swift.mp3\", n_channels=2)\n",
"test_input_audio = input_audio.get_segment(from_s=180, to_s=200.1)\n",
"test_input_audio.play(compress=False)"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "9a4b7802",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:53.049861Z",
"start_time": "2024-04-10T00:42:53.047587Z"
}
},
"outputs": [],
"source": [
"prior_texts = \"\"\"\n",
"[Intro]\n",
"Ah ah ha ha ha\n",
"Ah ah ha ha ha\n",
"Ah ah ha ha ha\n",
"Ah ah ha ha ha\n",
"Ah ah ha ha ha\n",
"\n",
"[Verse]\n",
"Been here all along\n",
"So, why can't you see?\n",
"You belong with me\n",
"Standing by and waiting at your backdoor\n",
"All this time how could you not know, baby?\n",
"You belong with me\n",
"You belong with me \n",
"\n",
"[Chorus]\n",
"You belong with me\n",
"You belong with me\n",
"You belong with me\n",
"You belong with me\n",
"\"\"\""
]
},
{
"cell_type": "code",
"execution_count": 8,
"id": "8453ea7f",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:42:56.384421Z",
"start_time": "2024-04-10T00:42:53.650577Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"with prior --> here all along so why can't you see you belong with me standing by you waiting at your backdo all this time how could you not know baby\n",
"without prior --> here all along so why can't you see you belong with me standing by you waiting at your back door all this time how could you not know baby\n"
]
}
],
"source": [
"hoot_without_prior = encode(test_input_audio)\n",
"hoot_with_prior = encode(test_input_audio, prior_texts=[prior_texts])\n",
"print(\"with prior -->\", hoot_with_prior)\n",
"print(\"without prior -->\", hoot_without_prior)"
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "21cbeb56",
"metadata": {
"ExecuteTime": {
"end_time": "2024-04-10T00:43:15.966254Z",
"start_time": "2024-04-10T00:43:15.754619Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"/home/tony/Work/tony/Preference\r\n"
]
}
],
"source": [
"!pwd"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "f5bb6a1a",
"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
}