Spaces:
Running
Running
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 |
-
|
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:
|