Gagan Bhatia commited on
Commit
c28c196
1 Parent(s): 744530a

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -365,6 +365,7 @@ class Summarization:
365
  f"{model_dir}", return_dict=True
366
  )
367
  elif model_type == "byt5":
 
368
 
369
  if use_gpu:
370
  if torch.cuda.is_available():
 
365
  f"{model_dir}", return_dict=True
366
  )
367
  elif model_type == "byt5":
368
+ self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_dir}")
369
 
370
  if use_gpu:
371
  if torch.cuda.is_available():