Spaces:
Runtime error
Runtime error
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 |
)
|