mirror of
https://github.com/mudler/LocalAI.git
synced 2024-06-07 19:40:48 +00:00
fb0a4c5d9a
* Dockerfile changes to build for ROCm * Adjust linker flags for ROCm * Update conda env for diffusers and transformers to use ROCm pytorch * Update transformers conda env for ROCm * ci: build hipblas images * fixup rebase * use self-hosted Signed-off-by: mudler <mudler@localai.io> * specify LD_LIBRARY_PATH only when BUILD_TYPE=hipblas --------- Signed-off-by: mudler <mudler@localai.io> Co-authored-by: mudler <mudler@localai.io>
15 lines
307 B
Makefile
15 lines
307 B
Makefile
CONDA_ENV_PATH = "transformers.yml"
|
|
|
|
ifeq ($(BUILD_TYPE), cublas)
|
|
CONDA_ENV_PATH = "transformers-nvidia.yml"
|
|
endif
|
|
|
|
ifeq ($(BUILD_TYPE), hipblas)
|
|
CONDA_ENV_PATH = "transformers-rocm.yml"
|
|
endif
|
|
|
|
.PHONY: transformers
|
|
transformers:
|
|
@echo "Installing $(CONDA_ENV_PATH)..."
|
|
bash install.sh $(CONDA_ENV_PATH)
|