Update README.md
Browse files
README.md
CHANGED
|
@@ -41,6 +41,7 @@ TBD
|
|
| 41 |
TBD
|
| 42 |
|
| 43 |
#### Training process
|
|
|
|
| 44 |
```python
|
| 45 |
import torch
|
| 46 |
from transformers import WhisperForConditionalGeneration, AutoProcessor, AutoTokenizer, AutoConfig
|
|
@@ -153,6 +154,7 @@ trainer.train()
|
|
| 153 |
```
|
| 154 |
|
| 155 |
### Use model for inference (with generate)
|
|
|
|
| 156 |
```python
|
| 157 |
from transformers import TextStreamer
|
| 158 |
|
|
|
|
| 41 |
TBD
|
| 42 |
|
| 43 |
#### Training process
|
| 44 |
+
!!! Make sure to install Transformers 4.46.0 !!!
|
| 45 |
```python
|
| 46 |
import torch
|
| 47 |
from transformers import WhisperForConditionalGeneration, AutoProcessor, AutoTokenizer, AutoConfig
|
|
|
|
| 154 |
```
|
| 155 |
|
| 156 |
### Use model for inference (with generate)
|
| 157 |
+
!!! Make sure to install Transformers 4.46.0 !!!
|
| 158 |
```python
|
| 159 |
from transformers import TextStreamer
|
| 160 |
|