Wing Lian 790df757cb don't install xformers in for arm64 (#3359) ...

* install xformers in the base docker image

* install numba and numpy first

* set CUDA_HOME for xformers install

* Set cuda  home env

* don't install xformers by default on aarch64/arm64

2026-01-16 09:02:37 -05:00

7.0 KiB

Raw Blame History

View Raw