Update model.py

src/models/model.py

@@ -315,7 +315,8 @@ class Summarization:
         )
 
         self.T5Model = LightningModel(
-            tokenizer=self.tokenizer, model=self.model, output=outputdir
+            tokenizer=self.tokenizer, model=self.model, output=outputdir,
+            learning_rate=learning_rate, adam_epsilon=adam_epsilon
         )
 
         MLlogger = MLFlowLogger(experiment_name="Summarization",