Failing build of NemotronOCRV2 on arm64
#1
by kyuukyuutai - opened
Failing build of NemotronOCRV2 on arm64
kyuukyuutai changed discussion status to closed
kyuukyuutai changed discussion status to open
When building NemotronOCRV2 on arm64, the build fails with numerous errors c++: error: unrecognized command-line option '-mavx2'.
This stems from the incorrect detection of the plateform type, that adds x86_64 option even if you're on arm64.
To correct this, edit the file build-extension.py line76-7l7 and replace if os.environ.get("ARCH") != "arm64": by if platform.machine().lower() in ("x86_64", "amd64"):
I made a PR to fix this https://huggingface.co/nvidia/nemotron-ocr-v2/commit/fb83fbea91eed704063ff5479930e59705b41b4c should behave better for others who try to install on arm
emelryan changed discussion status to closed