Didier commited on
Commit
b929bff
1 Parent(s): fc9e3cb

Switched back to using MADLAD 3b model (from 7b) due to being GPU poor

Browse files
Files changed (1) hide show
  1. model_translation.py +1 -1
model_translation.py CHANGED
@@ -55,7 +55,7 @@ class ModelMADLAD(metaclass=Singleton):
55
  """Loads an instance of the Google MADLAD model (3B).
56
  """
57
  def __init__(self):
58
- self._model_name = "google/madlad400-7b-mt"
59
  self._tokenizer = AutoTokenizer.from_pretrained(
60
  self.model_name, use_fast=True
61
  )
 
55
  """Loads an instance of the Google MADLAD model (3B).
56
  """
57
  def __init__(self):
58
+ self._model_name = "google/madlad400-3b-mt"
59
  self._tokenizer = AutoTokenizer.from_pretrained(
60
  self.model_name, use_fast=True
61
  )