Gagan Bhatia commited on
Commit
08eb467
1 Parent(s): 8e6d41d

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +2 -0
src/models/model.py CHANGED
@@ -33,3 +33,5 @@ class DataModule(Dataset):
33
  target_max_token_len: int = 512,
34
  ):
35
  """
 
 
 
33
  target_max_token_len: int = 512,
34
  ):
35
  """
36
+ :param data:
37
+ :param tokenizer: