speechbrainteam commited on
Commit
1f024d3
1 Parent(s): 98d299b

Update README.md

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