Didier commited on
Commit
77364cc
·
verified ·
1 Parent(s): 46986c0

Update model_translation.py

Browse files
Files changed (1) hide show
  1. model_translation.py +2 -2
model_translation.py CHANGED
@@ -21,10 +21,10 @@ class Singleton(type):
21
  return cls._instances[cls]
22
 
23
  class ModelM2M100(metaclass=Singleton):
24
- """Loads an instance of the M2M100 model (418M).
25
  """
26
  def __init__(self):
27
- self._model_name = "facebook/m2m100_418M"
28
  self._tokenizer = M2M100Tokenizer.from_pretrained(self._model_name)
29
  self._model = M2M100ForConditionalGeneration.from_pretrained(
30
  self._model_name,
 
21
  return cls._instances[cls]
22
 
23
  class ModelM2M100(metaclass=Singleton):
24
+ """Loads an instance of the M2M100 model.
25
  """
26
  def __init__(self):
27
+ self._model_name = "facebook/m2m100_1.2B"
28
  self._tokenizer = M2M100Tokenizer.from_pretrained(self._model_name)
29
  self._model = M2M100ForConditionalGeneration.from_pretrained(
30
  self._model_name,