Spaces:
Runtime error
Runtime error
mekaneeky
commited on
Commit
•
a75129b
1
Parent(s):
d12549e
fix missing argument in hf_hub_download
Browse files- stitched_model.py +4 -3
stitched_model.py
CHANGED
@@ -38,10 +38,11 @@ class CombinedModel(nn.Module):
|
|
38 |
self.hps.data.filter_length // 2 + 1,
|
39 |
self.hps.train.segment_size // self.hps.data.hop_length,
|
40 |
**self.hps.model)
|
41 |
-
if
|
42 |
-
g_pth = f"{self.tts_checkpoint_path}/{self.tts_checkpoint_name}"
|
43 |
-
else:
|
44 |
g_pth = self.tts_checkpoint_path
|
|
|
|
|
|
|
45 |
_ = utils.load_checkpoint(g_pth, self.tts_synth, None)
|
46 |
|
47 |
|
|
|
38 |
self.hps.data.filter_length // 2 + 1,
|
39 |
self.hps.train.segment_size // self.hps.data.hop_length,
|
40 |
**self.hps.model)
|
41 |
+
if huggingface_checkpoint:
|
|
|
|
|
42 |
g_pth = self.tts_checkpoint_path
|
43 |
+
else:
|
44 |
+
g_pth = f"{self.tts_checkpoint_path}/{self.tts_checkpoint_name}"
|
45 |
+
|
46 |
_ = utils.load_checkpoint(g_pth, self.tts_synth, None)
|
47 |
|
48 |
|