kennethnym/infinifi
1
0
Fork 0
Code Issues 2 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat: change model used

Browse Source
This commit is contained in:
Kenneth
2024-07-20 12:42:35 +01:00
parent 1cc81f4da2
commit f1ecddc918
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
generate.py
View File

@@ -2,7 +2,7 @@ import torchaudio
from audiocraft.models.magnet import MAGNeT from audiocraft.models.magnet import MAGNeT
from audiocraft.data.audio import audio_write from audiocraft.data.audio import audio_write
MODEL_NAME = "facebook/magnet-small-10secs" MODEL_NAME = "facebook/magnet-medium-30secs"
print(f"getting {MODEL_NAME}...") print(f"getting {MODEL_NAME}...")