speechbrainteam commited on
Commit
c74c9bd
1 Parent(s): efab5aa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -51,6 +51,8 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
51
 
52
 
53
  ```
 
 
54
 
55
  #### Referencing SpeechBrain
56
 
@@ -61,7 +63,7 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
61
  year = {2021},
62
  publisher = {GitHub},
63
  journal = {GitHub repository},
64
- howpublished = {\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
65
  }
66
  ```
67
 
 
51
 
52
 
53
  ```
54
+ ### Inference on GPU
55
+ To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
56
 
57
  #### Referencing SpeechBrain
58
 
 
63
  year = {2021},
64
  publisher = {GitHub},
65
  journal = {GitHub repository},
66
+ howpublished = {\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
67
  }
68
  ```
69