Update parser.py

Former-commit-id: e90c424f55b17e4971f8b9d85b6aeac89bb6b98e
This commit is contained in:
hoshi-hiyouga 2024-06-24 21:37:42 +08:00 committed by GitHub
parent a9f10a9abd
commit e74fcdf7b1

View File

@ -102,10 +102,6 @@ def _verify_model_args(model_args: "ModelArguments", finetuning_args: "Finetunin
if model_args.adapter_name_or_path is not None and len(model_args.adapter_name_or_path) != 1: if model_args.adapter_name_or_path is not None and len(model_args.adapter_name_or_path) != 1:
raise ValueError("Quantized model only accepts a single adapter. Merge them first.") raise ValueError("Quantized model only accepts a single adapter. Merge them first.")
# In case that `flash_attn` is set to `off` in the yaml file, and parsed as `False` afterwards.
if model_args.flash_attn == False:
raise ValueError("flash_attn should be \"disabled\", \"sdpa\", \"fa2\" or \"auto\".")
def _check_extra_dependencies( def _check_extra_dependencies(
model_args: "ModelArguments", model_args: "ModelArguments",