Spaces:
Running
Running
Switched back to using MADLAD 3b model (from 7b) due to being GPU poor
Browse files- 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-
|
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 |
)
|