Didier commited on
Commit
150301e
1 Parent(s): 706408b

Update model_translation.py

Browse files
Files changed (1) hide show
  1. model_translation.py +1 -1
model_translation.py CHANGED
@@ -48,7 +48,7 @@ class ModelM2M100(metaclass=Singleton):
48
  class ModelMADLAD(metaclass=Singleton):
49
  """Loads an instance of the Google MADLAD model (3B).
50
  """
51
- def __init__(self, model_name):
52
  self._model_name = "google/madlad400-3b-mt"
53
  self._tokenizer = AutoTokenizer.from_pretrained(
54
  self.model_name, use_fast=True
 
48
  class ModelMADLAD(metaclass=Singleton):
49
  """Loads an instance of the Google MADLAD model (3B).
50
  """
51
+ def __init__(self):
52
  self._model_name = "google/madlad400-3b-mt"
53
  self._tokenizer = AutoTokenizer.from_pretrained(
54
  self.model_name, use_fast=True