Skip to content

Commit 644a500

Browse files
authored
use smaller portion of dataset (#902)
1 parent d918350 commit 644a500

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ class TestOneshotAndFinetune(unittest.TestCase):
1919
def _test_oneshot_and_finetune(self):
2020
from llmcompressor.transformers import apply
2121

22-
splits = {"train": "train[:50%]", "calibration": "train[50%:60%]"}
22+
splits = {"train": "train[:30%]", "calibration": "train[30%:40%]"}
2323
if self.dataset == "ultrachat-200k":
24-
splits = {"train": "train_gen[:50%]", "calibration": "train_gen[50%:60%]"}
24+
splits = {"train": "train_gen[:30%]", "calibration": "train_gen[30%:40%]"}
2525

2626
apply(
2727
model=self.model,

0 commit comments

Comments
 (0)