mrprimenotes
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"
|