pc

mosaicml · Jun 13, 2024 · 4580bf8 · 4580bf8
1 parent 08a505d
commit 4580bf8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,
         )