Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
a51b600
1
Parent(s):
10396f5
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -6,6 +6,7 @@ from transformers import (
|
|
| 6 |
T5TokenizerFast as T5Tokenizer,
|
| 7 |
MT5Tokenizer,
|
| 8 |
MT5ForConditionalGeneration,
|
|
|
|
| 9 |
)
|
| 10 |
from torch.utils.data import Dataset, DataLoader
|
| 11 |
import pytorch_lightning as pl
|
|
|
|
| 6 |
T5TokenizerFast as T5Tokenizer,
|
| 7 |
MT5Tokenizer,
|
| 8 |
MT5ForConditionalGeneration,
|
| 9 |
+
ByT5Tokenizer,
|
| 10 |
)
|
| 11 |
from torch.utils.data import Dataset, DataLoader
|
| 12 |
import pytorch_lightning as pl
|