from local

This commit is contained in:
dohe0342 2023-06-09 17:31:30 +09:00
parent 2685e8f702
commit 7613ed5262
2 changed files with 3 additions and 2 deletions

View File

@ -59,8 +59,9 @@ from beam_search import (
greedy_search_batch,
modified_beam_search,
)
#from train import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
from prompt_tuning import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
from train_tta import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
#from prompt_tuning import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
from icefall.checkpoint import (
average_checkpoints,