tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Clean

Browse Source
This commit is contained in:
litagin02
2024-02-07 10:36:36 +09:00
parent 0d7de644cd
commit 27518345dd
2 changed files with 2 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@@ -25,3 +25,5 @@ venv/
*.csv *.csv
*.bak *.bak
/mos_results/ /mos_results/
safetensors.ipynb

115
safetensors.ipynb
View File

@@ -1,115 +0,0 @@
{
"cells": [
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
"import torch\n",
"\n",
"model_dir = \"pretrained-Japanese-Extra\"\n",
"step = 0\n",
"\n",
"g_model = torch.load(f\"{model_dir}/G_{step}.pth\", map_location=\"cpu\")\n",
"d_model = torch.load(f\"{model_dir}/D_{step}.pth\", map_location=\"cpu\")\n",
"wd_model = torch.load(f\"{model_dir}/WD_{step}.pth\", map_location=\"cpu\")"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"skip enc_p.in_feature_net.0.weight\n",
"skip enc_p.in_feature_net.2.weight\n",
"skip enc_p.in_feature_net.2.bias\n",
"skip enc_p.in_feature_net.3.norm.weight\n",
"skip enc_p.in_feature_net.3.norm.bias\n",
"skip enc_p.in_feature_net.3.mlp.c_fc1.weight\n",
"skip enc_p.in_feature_net.3.mlp.c_fc2.weight\n",
"skip enc_p.in_feature_net.3.mlp.c_proj.weight\n",
"skip enc_p.in_feature_net.4.weight\n",
"skip enc_p.emo_vq._codebook.initted\n",
"skip enc_p.emo_vq._codebook.cluster_size\n",
"skip enc_p.emo_vq._codebook.embed_avg\n",
"skip enc_p.emo_vq._codebook.embed\n",
"skip enc_p.out_feature_net.weight\n",
"skip enc_p.out_feature_net.bias\n"
]
}
],
"source": [
"g_dict = {}\n",
"skip_list = [\"enc_p.in_feature_net.\", \"enc_p.emo_vq.\", \"enc_p.out_feature_net.\"]\n",
"for key in g_model[\"model\"].keys():\n",
" if any([key.startswith(s) for s in skip_list]):\n",
" print(f\"skip {key}\")\n",
" continue\n",
" else:\n",
" g_dict[key] = g_model[\"model\"][key]"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [],
"source": [
"d_dict = {}\n",
"for key in d_model[\"model\"].keys():\n",
" d_dict[key] = d_model[\"model\"][key]"
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {},
"outputs": [],
"source": [
"wd_dict = {}\n",
"for key in wd_model[\"model\"].keys():\n",
" wd_dict[key] = wd_model[\"model\"][key]"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [],
"source": [
"from safetensors.torch import save_file\n",
"\n",
"\n",
"save_file(g_dict, f\"G_0.safetensors\")\n",
"save_file (d_dict, f\"D_0.safetensors\")\n",
"save_file (wd_dict, f\"WD_0.safetensors\")"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "venv",
"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.11"
}
},
"nbformat": 4,
"nbformat_minor": 2
}