Commit
•
820763d
1
Parent(s):
c3e5a3b
fix: typo
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def load_models():
|
|
16 |
|
17 |
for call_name, real_name in model_name_dict.items():
|
18 |
print("\tLoading model: %s" % call_name)
|
19 |
-
model =
|
20 |
tokenizer = AutoTokenizer.from_pretrained(real_name)
|
21 |
model_dict[call_name + "_model"] = model
|
22 |
model_dict[call_name + "_tokenizer"] = tokenizer
|
|
|
16 |
|
17 |
for call_name, real_name in model_name_dict.items():
|
18 |
print("\tLoading model: %s" % call_name)
|
19 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(real_name)
|
20 |
tokenizer = AutoTokenizer.from_pretrained(real_name)
|
21 |
model_dict[call_name + "_model"] = model
|
22 |
model_dict[call_name + "_tokenizer"] = tokenizer
|