switch -> git+ url

This commit is contained in:
mrfakename
2025-05-06 20:19:51 -07:00
parent 38d62623fc
commit afa0332e8c
6 changed files with 16 additions and 14 deletions
+2 -2
View File
@@ -164,5 +164,5 @@ if __name__ == "__main__":
print("latents shape: ", latents.shape)
print("latent_lengths: ", latent_lengths)
print("sr: ", sr)
torchaudio.save("test_reconstructed.flac", pred_wavs[0], sr)
print("test_reconstructed.flac")
torchaudio.save("test_reconstructed.wav", pred_wavs[0], sr)
print("test_reconstructed.wav")
+2 -2
View File
@@ -574,7 +574,7 @@ class ADaMoSHiFiGANV1(ModelMixin, ConfigMixin, FromOriginalModelMixin):
if __name__ == "__main__":
import soundfile as sf
x = "test_audio.flac"
x = "test_audio.wav"
model = ADaMoSHiFiGANV1.from_pretrained(
"./checkpoints/music_vocoder", local_files_only=True
)
@@ -584,4 +584,4 @@ if __name__ == "__main__":
mel = model.encode(wav)
wav = model.decode(mel)[0].mT
sf.write("test_audio_vocoder_rec.flac", wav.cpu().numpy(), 44100)
sf.write("test_audio_vocoder_rec.wav", wav.cpu().numpy(), 44100)