Spaces:
Sleeping
Sleeping
Hugo Flores Garcia
commited on
Commit
•
7f524ea
1
Parent(s):
980f2e7
lora ckpts aren't working?!
Browse files- scripts/exp/fine_tune.py +0 -2
scripts/exp/fine_tune.py
CHANGED
|
@@ -49,10 +49,8 @@ def fine_tune(audio_files_or_folders: List[str], name: str):
|
|
| 49 |
|
| 50 |
interface_conf = {
|
| 51 |
"Interface.coarse_ckpt": f"./runs/{name}/coarse/latest/vampnet/weights.pth",
|
| 52 |
-
"Interface.coarse_lora_ckpt": f"./runs/{name}/coarse/latest/lora.pth",
|
| 53 |
|
| 54 |
"Interface.coarse2fine_ckpt": f"./runs/{name}/c2f/latest/vampnet/weights.pth",
|
| 55 |
-
"Interface.coarse2fine_lora_ckpt": f"./runs/{name}/c2f/latest/lora.pth",
|
| 56 |
"Interface.wavebeat_ckpt": "./models/wavebeat.pth",
|
| 57 |
|
| 58 |
"Interface.codec_ckpt": "./models/vampnet/codec.pth",
|
|
|
|
| 49 |
|
| 50 |
interface_conf = {
|
| 51 |
"Interface.coarse_ckpt": f"./runs/{name}/coarse/latest/vampnet/weights.pth",
|
|
|
|
| 52 |
|
| 53 |
"Interface.coarse2fine_ckpt": f"./runs/{name}/c2f/latest/vampnet/weights.pth",
|
|
|
|
| 54 |
"Interface.wavebeat_ckpt": "./models/wavebeat.pth",
|
| 55 |
|
| 56 |
"Interface.codec_ckpt": "./models/vampnet/codec.pth",
|