hiyouga / LLaMA-Factory

Unify Efficient Fine-Tuning of 100+ LLMs

Apache License 2.0

25.26k stars 3.13k forks source link

Bug Fix: `off` is parsed as `False` in yaml file #4446

Closed stceum closed 4 days ago

stceum commented 4 days ago

Bug Fix: off is parsed as False in yaml file, changed to disabled to avoid this.

What does this PR do?

Fixes # (issue)

I have tried to set flash_attn to off, and got error Unknown attention type: False. Then this bug is found and fixed as submitted.

Reference: [1] https://github.com/yaml/pyyaml/issues/376

Before submitting

[x] Did you read the contributor guideline?
[ ] Did you write any new necessary tests?