File tree 2 files changed +7
-7
lines changed
test/3x/torch/quantization/weight_only
2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -238,13 +238,13 @@ def test_mllm(self):
238
238
image_processor = None ,
239
239
dataset = "liuhaotian/llava_conv_58k" ,
240
240
extra_data_dir = None ,
241
- seqlen = 512 ,
241
+ seqlen = 32 ,
242
242
batch_size = 1 ,
243
243
split = None ,
244
244
apply_template = None ,
245
245
truncation = False ,
246
246
seed = 42 ,
247
- nsamples = 5 ,
247
+ nsamples = 1 ,
248
248
gradient_accumulate_steps = 1 ,
249
249
quant_nontext_module = False ,
250
250
processor = processor ,
@@ -253,9 +253,9 @@ def test_mllm(self):
253
253
bits = 4 ,
254
254
group_size = 128 ,
255
255
is_mllm = True ,
256
- nsamples = 5 ,
256
+ nsamples = 1 ,
257
257
batch_size = batch_size ,
258
- iters = 2 ,
258
+ iters = 1 ,
259
259
seqlen = seqlen ,
260
260
quant_nontext_module = False ,
261
261
truncation = truncation ,
Original file line number Diff line number Diff line change @@ -227,9 +227,9 @@ def test_vlm(self):
227
227
group_size = 128 ,
228
228
is_vlm = True ,
229
229
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 ,
233
233
batch_size = 1 ,
234
234
)
235
235
You can’t perform that action at this time.
0 commit comments