Commit 1d79ed9b authored by Angela Fan's avatar Angela Fan
Browse files

adding model arg override at generation time for interactive.py

parent 30ef667d

interactive.py

+1 −1
Changes for interactive.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -73,7 +73,7 @@ def main(args): 
    # Load ensemble

    print('| loading model(s) from {}'.format(args.path))

    model_paths = args.path.split(':')

    models, model_args = utils.load_ensemble_for_inference(model_paths, task)

    models, model_args = utils.load_ensemble_for_inference(model_paths, task, model_arg_overrides=eval(args.model_overrides))



    # Set dictionaries

    src_dict = task.source_dictionary