lshzhm commited on
Commit
b61a5a8
·
1 Parent(s): 1260986

Update infer_cli_test.py

Browse files
F5-TTS/src/f5_tts/infer/infer_cli_test.py CHANGED
@@ -484,7 +484,7 @@ if __name__ == "__main__":
484
  energy_p = energy_p / max(energy_p)
485
  energy_p = torch.from_numpy(energy_p).unsqueeze(0).unsqueeze(2)
486
 
487
- #print("energy shape", energy_p.shape, energy.shape)
488
  #energy = torch.cat([energy_p, energy], dim=1)
489
 
490
  try:
 
484
  energy_p = energy_p / max(energy_p)
485
  energy_p = torch.from_numpy(energy_p).unsqueeze(0).unsqueeze(2)
486
 
487
+ print("############energy shape", energy_p.shape, energy.shape, type(energy_p), type(energy), energy_p.dtype, energy.dtype)
488
  #energy = torch.cat([energy_p, energy], dim=1)
489
 
490
  try: