Christopher H.
commited on
Update Readme
Browse files
README.md
CHANGED
|
@@ -44,7 +44,7 @@ TBD
|
|
| 44 |
!!! Make sure to install Transformers 4.46.0 !!!
|
| 45 |
```python
|
| 46 |
import torch
|
| 47 |
-
from transformers import WhisperForConditionalGeneration, AutoProcessor, AutoTokenizer, AutoConfig
|
| 48 |
from datasets import load_dataset
|
| 49 |
|
| 50 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 44 |
!!! Make sure to install Transformers 4.46.0 !!!
|
| 45 |
```python
|
| 46 |
import torch
|
| 47 |
+
from transformers import WhisperForConditionalGeneration, AutoProcessor, AutoTokenizer, AutoConfig, TextStreamer, Trainer
|
| 48 |
from datasets import load_dataset
|
| 49 |
|
| 50 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|