Gagan Bhatia commited on
Commit
9f5c468
·
1 Parent(s): 4fd5b35

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +0 -1
src/models/model.py CHANGED
@@ -152,7 +152,6 @@ class PLDataModule(LightningDataModule):
152
  class LightningModel(LightningModule):
153
  """ PyTorch Lightning Model class"""
154
 
155
- def __init__(self, tokenizer, model, learning_rate, adam_epsilon, output: str = "outputs"):
156
  """
157
  initiates a PyTorch Lightning Model
158
  Args:
 
152
  class LightningModel(LightningModule):
153
  """ PyTorch Lightning Model class"""
154
 
 
155
  """
156
  initiates a PyTorch Lightning Model
157
  Args: