This commit is contained in:
hiyouga committed 2023-11-28 20:52:28 +08:00
1 parent c2d4300ac4
commit 475a3fa0f4
4 files changed
+12 -6

No files matched your search

+3
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.")