We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d918350 commit 644a500Copy full SHA for 644a500
tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py
@@ -19,9 +19,9 @@ class TestOneshotAndFinetune(unittest.TestCase):
19
def _test_oneshot_and_finetune(self):
20
from llmcompressor.transformers import apply
21
22
- splits = {"train": "train[:50%]", "calibration": "train[50%:60%]"}
+ splits = {"train": "train[:30%]", "calibration": "train[30%:40%]"}
23
if self.dataset == "ultrachat-200k":
24
- splits = {"train": "train_gen[:50%]", "calibration": "train_gen[50%:60%]"}
+ splits = {"train": "train_gen[:30%]", "calibration": "train_gen[30%:40%]"}
25
26
apply(
27
model=self.model,
0 commit comments