support cuda

Files changed (2) hide show

README.md CHANGED Viewed

@@ -19,7 +19,8 @@ See `requirements.txt`
 ```python
 from transformers import AutoModel
-model = AutoModel.from_pretrained("hance-ai/audiomae", trust_remote_code=True)  # load the pretrained model
 z = model('path/audio_fname.wav')  # (768, 8, 64) = (latent_dim_size, latent_freq_dim, latent_temporal_dim)
 ```

 ```python
 from transformers import AutoModel
+device = 'cpu'  # 'cpu' or 'cuda'
+model = AutoModel.from_pretrained("hance-ai/audiomae", trust_remote_code=True).to(device)  # load the pretrained model
 z = model('path/audio_fname.wav')  # (768, 8, 64) = (latent_dim_size, latent_freq_dim, latent_temporal_dim)
 ```

save_audioMAE_self_sustainable.ipynb CHANGED Viewed

The diff for this file is too large to render. See raw diff