hynt commited on
Commit
8e7a4f5
·
verified ·
1 Parent(s): 00d7219

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +7 -7
model.py CHANGED
@@ -377,38 +377,38 @@ def _get_vietnamese_pretrained_model(
377
  # subfolder=".",
378
  # )
379
 
380
- decoder_model = "decoder-epoch-45-avg-25.onnx"
381
 
382
- if repo_id == "csukuangfj/sherpa-onnx-zipformer-vi-int8-2025-04-20":
383
  # encoder_model = _get_nn_model_filename(
384
  # repo_id=repo_id,
385
  # filename="encoder-epoch-12-avg-8.int8.onnx",
386
  # subfolder=".",
387
  # )
388
 
389
- encoder_model = "encoder-epoch-45-avg-25.int8.onnx"
390
 
391
  # joiner_model = _get_nn_model_filename(
392
  # repo_id=repo_id,
393
  # filename="joiner-epoch-12-avg-8.int8.onnx",
394
  # subfolder=".",
395
  # )
396
- joiner_model = "joiner-epoch-45-avg-25.int8.onnx"
397
- elif repo_id == "csukuangfj/sherpa-onnx-zipformer-vi-2025-04-20":
398
  # encoder_model = _get_nn_model_filename(
399
  # repo_id=repo_id,
400
  # filename="encoder-epoch-12-avg-8.onnx",
401
  # subfolder=".",
402
  # )
403
 
404
- encoder_model = "encoder-epoch-45-avg-25.onnx"
405
 
406
  # joiner_model = _get_nn_model_filename(
407
  # repo_id=repo_id,
408
  # filename="joiner-epoch-12-avg-8.onnx",
409
  # subfolder=".",
410
  # )
411
- joiner_model = "joiner-epoch-45-avg-25.onnx"
412
  else:
413
  raise ValueError(f"repo_id: {repo_id}")
414
 
 
377
  # subfolder=".",
378
  # )
379
 
380
+ decoder_model = "decoder-epoch-20-avg-10.onnx"
381
 
382
+ if repo_id == "hynt/sherpa-onnx-zipformer-vi-int8-2025-10-16":
383
  # encoder_model = _get_nn_model_filename(
384
  # repo_id=repo_id,
385
  # filename="encoder-epoch-12-avg-8.int8.onnx",
386
  # subfolder=".",
387
  # )
388
 
389
+ encoder_model = "encoder-epoch-20-avg-10.int8.onnx"
390
 
391
  # joiner_model = _get_nn_model_filename(
392
  # repo_id=repo_id,
393
  # filename="joiner-epoch-12-avg-8.int8.onnx",
394
  # subfolder=".",
395
  # )
396
+ joiner_model = "joiner-epoch-20-avg-10.int8.onnx"
397
+ elif repo_id == "hynt/sherpa-onnx-zipformer-vi-2025-10-16":
398
  # encoder_model = _get_nn_model_filename(
399
  # repo_id=repo_id,
400
  # filename="encoder-epoch-12-avg-8.onnx",
401
  # subfolder=".",
402
  # )
403
 
404
+ encoder_model = "encoder-epoch-20-avg-10.onnx"
405
 
406
  # joiner_model = _get_nn_model_filename(
407
  # repo_id=repo_id,
408
  # filename="joiner-epoch-12-avg-8.onnx",
409
  # subfolder=".",
410
  # )
411
+ joiner_model = "joiner-epoch-20-avg-10.onnx"
412
  else:
413
  raise ValueError(f"repo_id: {repo_id}")
414