speechbrainteam commited on
Commit
abdf473
1 Parent(s): bb56d83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -62,6 +62,8 @@ asr_model = EncoderDecoderASR.from_hparams(source="speechbrain/asr-wav2vec2-comm
62
  asr_model.transcribe_file("example-fr.wav")
63
 
64
  ```
 
 
65
 
66
  #### Referencing SpeechBrain
67
 
@@ -72,6 +74,6 @@ asr_model.transcribe_file("example-fr.wav")
72
  year = {2021},
73
  publisher = {GitHub},
74
  journal = {GitHub repository},
75
- howpublished = {\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
76
  }
77
  ```
 
62
  asr_model.transcribe_file("example-fr.wav")
63
 
64
  ```
65
+ ### Inference on GPU
66
+ To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
67
 
68
  #### Referencing SpeechBrain
69
 
 
74
  year = {2021},
75
  publisher = {GitHub},
76
  journal = {GitHub repository},
77
+ howpublished = {\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\url{https://github.com/speechbrain/speechbrain}},
78
  }
79
  ```