csukuangfj commited on
Commit
82c3ef1
1 Parent(s): c9b8243

add arabic models

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -418,7 +418,7 @@ def _get_arabic_pre_trained_model(
418
 
419
  nn_model = _get_nn_model_filename(
420
  repo_id=repo_id,
421
- filename=filename,
422
  )
423
 
424
  tokens = _get_token_filename(repo_id=repo_id, subfolder="data/lang_bpe_5000")
 
418
 
419
  nn_model = _get_nn_model_filename(
420
  repo_id=repo_id,
421
+ filename="cpu_jit.pt",
422
  )
423
 
424
  tokens = _get_token_filename(repo_id=repo_id, subfolder="data/lang_bpe_5000")