Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
168d045
1
Parent(s):
08eb467
Update model.py
Browse files- src/models/model.py +2 -0
src/models/model.py
CHANGED
@@ -35,3 +35,5 @@ class DataModule(Dataset):
|
|
35 |
"""
|
36 |
:param data:
|
37 |
:param tokenizer:
|
|
|
|
|
|
35 |
"""
|
36 |
:param data:
|
37 |
:param tokenizer:
|
38 |
+
:param source_max_token_len:
|
39 |
+
:param target_max_token_len:
|