Gagan Bhatia commited on
Commit
5866066
1 Parent(s): bcf1243

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -1
src/models/model.py CHANGED
@@ -385,7 +385,7 @@ class Summarization:
385
  trainer.fit(self.T5Model, self.data_module)
386
 
387
  def load_model(
388
- self, model_type: str = 't5', model_dir: str = "models", use_gpu: bool = False
389
  ):
390
  """
391
  loads a checkpoint for inferencing/prediction
 
385
  trainer.fit(self.T5Model, self.data_module)
386
 
387
  def load_model(
388
+ self, model_type: str = "t5", model_dir: str = "models", use_gpu: bool = False
389
  ):
390
  """
391
  loads a checkpoint for inferencing/prediction