Spaces:

breadlicker45
/

WavTokenizer-demo

Build error

breadlicker45 commited on Sep 9, 2024

Commit

21756e2

verified ·

1 Parent(s): b2ee94f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from decoder.pretrained import WavTokenizer
 # Initialize WavTokenizer
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-config_path = "configs/wavtokenizer_smalldata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
 model_path = "wavtokenizer_medium_music_audio_320_24k.ckpt"
 wavtokenizer = WavTokenizer.from_pretrained0802(config_path, model_path)

 # Initialize WavTokenizer
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+config_path = "wavtokenizer_mediumdata_music_audio_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
 model_path = "wavtokenizer_medium_music_audio_320_24k.ckpt"
 wavtokenizer = WavTokenizer.from_pretrained0802(config_path, model_path)