speechbrainteam commited on
Commit
db19853
1 Parent(s): 6b8b275

Update README.md

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