Commit
·
6c15050
1
Parent(s):
fa00390
update model to torch 1.10.0
Browse files
model.py
CHANGED
@@ -243,7 +243,7 @@ def _get_tibetan_pre_trained_model(repo_id: str):
|
|
243 |
|
244 |
nn_model_filename = _get_nn_model_filename(
|
245 |
repo_id=repo_id,
|
246 |
-
filename="cpu_jit.pt",
|
247 |
)
|
248 |
bpe_model_filename = _get_bpe_model_filename(repo_id=repo_id)
|
249 |
|
|
|
243 |
|
244 |
nn_model_filename = _get_nn_model_filename(
|
245 |
repo_id=repo_id,
|
246 |
+
filename="cpu_jit-epoch-28-avg-23-torch-1.10.0.pt",
|
247 |
)
|
248 |
bpe_model_filename = _get_bpe_model_filename(repo_id=repo_id)
|
249 |
|