Skip to content

Commit 7aeeeca

Browse files
committed
reduce vlm ut
Signed-off-by: Kaihui-intel <kaihui.tang@intel.com>
1 parent daec2cd commit 7aeeeca

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

test/3x/torch/quantization/weight_only/test_autoround.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -238,13 +238,13 @@ def test_mllm(self):
238238
image_processor=None,
239239
dataset="liuhaotian/llava_conv_58k",
240240
extra_data_dir=None,
241-
seqlen=512,
241+
seqlen=32,
242242
batch_size=1,
243243
split=None,
244244
apply_template=None,
245245
truncation=False,
246246
seed=42,
247-
nsamples=5,
247+
nsamples=1,
248248
gradient_accumulate_steps=1,
249249
quant_nontext_module=False,
250250
processor=processor,
@@ -253,9 +253,9 @@ def test_mllm(self):
253253
bits=4,
254254
group_size=128,
255255
is_mllm=True,
256-
nsamples=5,
256+
nsamples=1,
257257
batch_size=batch_size,
258-
iters=2,
258+
iters=1,
259259
seqlen=seqlen,
260260
quant_nontext_module=False,
261261
truncation=truncation,

test/3x/torch/quantization/weight_only/test_transformers.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -227,9 +227,9 @@ def test_vlm(self):
227227
group_size=128,
228228
is_vlm=True,
229229
dataset="NeelNanda/pile-10k",
230-
iters=2,
231-
n_samples=5,
232-
seq_len=512,
230+
iters=1,
231+
n_samples=1,
232+
seq_len=32,
233233
batch_size=1,
234234
)
235235

0 commit comments

Comments
 (0)