speechbrainteam commited on
Commit
bfb8913
1 Parent(s): a256fa8

Update README.md

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