mlc-llm icon indicating copy to clipboard operation
mlc-llm copied to clipboard

ARM64 CPU on Linux

Open Crystalxd opened this issue 1 year ago • 2 comments

Can mlc-llm precisely run arm64v8 CPU on Ubuntu? Not just on the Mac M series chip.

Crystalxd avatar Jun 09 '23 09:06 Crystalxd

detail: I can't install TVM by running pip install -I mlc_ai_nightly -f https://mlc.ai/wheels on my Ubuntu server, so I compile one by source code(https://github.com/apache/tvm), but when I run python build.py --hf-path=databricks/dolly-v2-3b I get an error about 'ImportError: cannot import name 'relax' from 'tvm''

Crystalxd avatar Jun 09 '23 09:06 Crystalxd

Seems that you are not using the correct TVM branch. Would you mind double checking: https://mlc.ai/mlc-llm/docs/install/tvm.html#id2

junrushao avatar Jun 09 '23 13:06 junrushao

Yes, it works for me according to the docs, thanks for much!

Crystalxd avatar Jun 13 '23 02:06 Crystalxd