diff --git a/textattack/attack_args.py b/textattack/attack_args.py index c33cc26b..3521ecc8 100644 --- a/textattack/attack_args.py +++ b/textattack/attack_args.py @@ -708,6 +708,7 @@ def _create_attack_from_args(cls, args, model_wrapper): if args.query_budget: recipe.goal_function.query_budget = args.query_budget recipe.goal_function.model_cache_size = args.model_cache_size + recipe.goal_function.batch_size = args.model_batch_size recipe.constraint_cache_size = args.constraint_cache_size return recipe elif args.attack_from_file: