Update model.py

src/models/model.py

@@ -40,3 +40,5 @@ class DataModule(Dataset):
         """
         self.data = data
         self.target_max_token_len = target_max_token_len
+        self.source_max_token_len = source_max_token_len
+        self.tokenizer = tokenizer