merge-weights.py: unsupported operand type(s) for |=: 'dict' and 'dict'

[paxvinci]

When I run python3 merge-weights.py --input_dir /home//llama-cpu/ --model_size 7B I receive the following error:

Traceback (most recent call last):
  File "merge-weights.py", line 168, in <module>
    main()
  File "merge-weights.py", line 161, in main
    write_model(
  File "merge-weights.py", line 63, in write_model
    state_dict |= {
TypeError: unsupported operand type(s) for |=: 'dict' and 'dict'

In the README I understood that it is necessary to run example-chat-bfloat16.py