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

adding model arg override at generation time for interactive.py

parent 30ef667d
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment