mrprimenotes commited on
Commit
40e7e0b
·
verified ·
1 Parent(s): a08b5b3

Update Readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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"