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 911ccd3 commit 960bd2bCopy full SHA for 960bd2b
neural_compressor/adaptor/torch_utils/gptq.py
@@ -248,7 +248,7 @@ def __init__(
248
249
# dataloader
250
self.use_max_length = use_max_length
251
- self.pad_max_length = pad_max_length
+ self.pad_max_length = pad_max_length or 2048
252
self.dataloader_original = dataloader
253
self.dataloader = []
254
self.nsamples = nsamples
0 commit comments