mekaneeky commited on
Commit
a75129b
1 Parent(s): d12549e

fix missing argument in hf_hub_download

Browse files
Files changed (1) hide show
  1. 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 not huggingface_checkpoint:
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