报错assert not torch.allclose(first_weight_old, first_weight)

报错:

Traceback (most recent call last):
  File "/home/bingxing2/ailab/group/ai4agr/wzf/LLM/models/Chinese-LLaMA-Alpaca/scripts/merge_llama_with_chinese_lora.py", line 327, in <module>
    assert not torch.allclose(first_weight_old, first_weight)
AssertionError

原因:peft版本问题

解决办法:

pip uninstall peft
pip install git+https://github.com/huggingface/peft.git@13e53fc