Skip to content

Commit 960bd2b

Browse files
authored
Fix case where pad_max_length set to None (#1729)
Signed-off-by: Ella Charlaix <ella@huggingface.co>
1 parent 911ccd3 commit 960bd2b

File tree

1 file changed

+1
-1
lines changed
  • neural_compressor/adaptor/torch_utils

1 file changed

+1
-1
lines changed

neural_compressor/adaptor/torch_utils/gptq.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def __init__(
248248

249249
# dataloader
250250
self.use_max_length = use_max_length
251-
self.pad_max_length = pad_max_length
251+
self.pad_max_length = pad_max_length or 2048
252252
self.dataloader_original = dataloader
253253
self.dataloader = []
254254
self.nsamples = nsamples

0 commit comments

Comments
 (0)