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): ...@@ -73,7 +73,7 @@ def main(args):
# Load ensemble # Load ensemble
print('| loading model(s) from {}'.format(args.path)) print('| loading model(s) from {}'.format(args.path))
model_paths = args.path.split(':') 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 # Set dictionaries
src_dict = task.source_dictionary 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