Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
8e6d41d
1
Parent(s):
5c300b0
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -30,3 +30,6 @@ class DataModule(Dataset):
|
|
| 30 |
data: pd.DataFrame,
|
| 31 |
tokenizer: T5Tokenizer,
|
| 32 |
source_max_token_len: int = 512,
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
data: pd.DataFrame,
|
| 31 |
tokenizer: T5Tokenizer,
|
| 32 |
source_max_token_len: int = 512,
|
| 33 |
+
target_max_token_len: int = 512,
|
| 34 |
+
):
|
| 35 |
+
"""
|