simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

fix bug

Browse source
This commit is contained in:
qwopqwop200 2023-05-27 07:47:17 +09:00 • committed by GitHub
parent b4eda619d0
commit 874c9fd0ef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/quantization/basic_usage_gpt_xl.py
View file

@ -56,7 +56,7 @@ def main():
model.seqlen = model_config[key] model.seqlen = model_config[key]
break break
else: else:
logger.warning("can't get model's sequence length from model config, will set to 2048.") print("can't get model's sequence length from model config, will set to 2048.")
model.seqlen = 2048 model.seqlen = 2048
# load train dataset for quantize # load train dataset for quantize