From a8adcba79cd88ce26d94537430b3926e58a43678 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Thu, 13 Jun 2024 10:11:57 -0700 Subject: [PATCH] int cast --- llmfoundry/data/finetuning/tasks.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py index fe1b07b25c..9a0f680bd7 100644 --- a/llmfoundry/data/finetuning/tasks.py +++ b/llmfoundry/data/finetuning/tasks.py @@ -680,11 +680,7 @@ def __getitem__(self, idx: int) -> Dict[str, Any]: def state_dict(self, num_samples: int, from_beginning: bool) -> Dict[str, Any]: if self.packing_ratio is not None: - num_samples = self.packing_ratio * num_samples - - print('+'*30) - print(num_samples) - print('+'*30) + num_samples = int(self.packing_ratio * num_samples) return super().state_dict( num_samples=num_samples,