Former-commit-id: 475a3fa0f4
This commit is contained in:
hiyouga
2023-11-28 20:52:28 +08:00
parent 33dc25e24f
commit ae1048db6d
4 changed files with 12 additions and 6 deletions

View File

@@ -146,6 +146,9 @@ def load_model_and_tokenizer(
# Quantization configurations (using bitsandbytes library)
if model_args.quantization_bit is not None:
if getattr(config, "quantization_config", None):
raise ValueError("Remove `quantization_bit` if you are using a quantized model.")
if is_deepspeed_zero3_enabled():
raise ValueError("DeepSpeed ZeRO-3 is incompatible with quantization.")