From 4580bf80ae4c3040691e89a1dd47a667b0bb6645 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Thu, 13 Jun 2024 09:52:15 -0700 Subject: [PATCH] pc --- llmfoundry/data/finetuning/tasks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py index 67102200cb..9fce9a8957 100644 --- a/llmfoundry/data/finetuning/tasks.py +++ b/llmfoundry/data/finetuning/tasks.py @@ -683,7 +683,8 @@ def state_dict(self, num_samples: int, num_samples = self.packing_ratio * num_samples return super().state_dict( - num_samples=num_samples, from_beginning=from_beginning + num_samples=num_samples, + from_beginning=from_beginning, )