Merge pull request #735 from Falanke21/patch-1

Fixed a batch_size bug in attack_args.py
QData · Aug 28, 2023 · 2abb4dd · 2abb4dd
2 parents 00adb8a + 1ba3e16
commit 2abb4dd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: