tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add missing positional arg (#450)

Browse Source
This commit is contained in:
Wing Lian
2023-08-21 04:10:19 -04:00
committed by GitHub
parent 04a42b6db1
commit 58cf7e7fed
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/trainer.py
View File

@@ -184,7 +184,7 @@ class AxolotlTrainer(Trainer):
rank=self.args.process_index, rank=self.args.process_index,
batch_size=self.args.per_device_eval_batch_size, batch_size=self.args.per_device_eval_batch_size,
) )
return super()._get_eval_sampler() return super()._get_eval_sampler(eval_dataset)
def get_train_dataloader(self) -> Union[DataLoader, MultipackDistributedDataloader]: def get_train_dataloader(self) -> Union[DataLoader, MultipackDistributedDataloader]:
if self.args.sample_packing: if self.args.sample_packing: