run_demo.py catch an error:AttributeError: module 'fastllm_pytools.llm' has no attribute 'set_device_map'.

[bigmover]

I don't know what I can do to solve this! I refer to fastllm (github) but can't find anything helpful