Meet Patel
commited on
Commit
·
f30c061
1
Parent(s):
5fe8607
Update checkpoint path in load_models function to use the latest training run
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -33,7 +33,7 @@ else:
|
|
| 33 |
fp16 = False
|
| 34 |
|
| 35 |
|
| 36 |
-
def load_models(f0_condition=False, checkpoint="./runs/training_run/
|
| 37 |
fp16 = torch.cuda.is_available()
|
| 38 |
|
| 39 |
if checkpoint is not None:
|
|
|
|
| 33 |
fp16 = False
|
| 34 |
|
| 35 |
|
| 36 |
+
def load_models(f0_condition=False, checkpoint="./runs/training_run/epochs_2nd_00020.pth", config="./configs/presets/config_dit_mel_seed_uvit_whisper_small_wavenet.yml"):
|
| 37 |
fp16 = torch.cuda.is_available()
|
| 38 |
|
| 39 |
if checkpoint is not None:
|