Update MODEL_URL

descriptinc · Oct 5, 2021 · f87e0c5 · f87e0c5
1 parent 6357293
commit f87e0c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/scripts/VQGAN-CLIP/generate.py b/scripts/VQGAN-CLIP/generate.py
@@ -1022,9 +1022,7 @@ def resize_image(image, out_size):
     if args.audio_index and args.audio_frame_length and args.audio_hop_length:
         start = args.audio_hop_length * args.audio_index
         audio = audio[start : start + args.audio_frame_length]
-    embed = torch.from_numpy(
-        wav2clip.embed_audio(np.expand_dims(audio, axis=0), wav2clip_model)
-    ).to(device)
+    embed = torch.from_numpy(wav2clip.embed_audio(audio, wav2clip_model)).to(device)
     pMs = []
     pMs.append(Prompt(embed, float(1.0), float("-inf")).to(device))
 

diff --git a/wav2clip/__init__.py b/wav2clip/__init__.py
@@ -4,7 +4,7 @@
 from .model.encoder import ResNetExtractor
 
 
-MODEL_URL = "https://www.dropbox.com/s/t3ews3k65n3atnh/Wav2CLIP.pt?dl=1"
+MODEL_URL = "https://github.com/descriptinc/lyrebird-wav2clip/releases/download/v0.1.0-alpha/Wav2CLIP.pt"
 
 
 def get_model(device="cpu", pretrained=True, frame_length=None, hop_length=None):