cointegrated
commited on
Commit
·
928013e
1
Parent(s):
f44876d
change the loading args
Browse files- translation.py +1 -1
translation.py
CHANGED
@@ -100,7 +100,7 @@ def sentenize_with_fillers(text, splitter, fix_double_space=True, ignore_errors=
|
|
100 |
|
101 |
class Translator:
|
102 |
def __init__(self):
|
103 |
-
self.model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_URL, low_cpu_mem_usage=
|
104 |
if torch.cuda.is_available():
|
105 |
self.model.cuda()
|
106 |
self.tokenizer = NllbTokenizer.from_pretrained(MODEL_URL)
|
|
|
100 |
|
101 |
class Translator:
|
102 |
def __init__(self):
|
103 |
+
self.model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_URL, low_cpu_mem_usage=True)
|
104 |
if torch.cuda.is_available():
|
105 |
self.model.cuda()
|
106 |
self.tokenizer = NllbTokenizer.from_pretrained(MODEL_URL)
|