fix the kwarg

axolotl-ai-cloud · Nov 8, 2023 · 0050a02 · 0050a02
1 parent 0daed17
commit 0050a02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
@@ -234,7 +234,7 @@ def get_eval_dataloader(
             }
             if self.args.dataloader_prefetch_factor:
                 dataloader_params[
-                    "dataloader_prefetch_factor"
+                    "prefetch_factor"
                 ] = self.args.dataloader_prefetch_factor
 
             if isinstance(eval_sampler, BatchSampler):