mirror/LocalAI
1
0 Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-10-19 21:12:37 -04:00
Code Issues Packages Projects Releases Wiki Activity

Build docker container for ROCm (#1595)

Browse Source 
* 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>
This commit is contained in:
fenfirandmudler authored and GitHub committed 2024-02-16 15:08:50 +01:00
1 parent e690bf387a
commit fb0a4c5d9a
9 files changed
+233 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

backend/python/diffusers/Makefile
+6 -2
View File
@@ -1,4 +1,8 @@
CONDA_ENV_PATH = "diffusers.yml"
export CONDA_ENV_PATH = "diffusers.yml"
ifeq ($(BUILD_TYPE), hipblas)
export CONDA_ENV_PATH = "diffusers-rocm.yml"
endif
.PHONY: diffusers
diffusers:
@@ -12,4 +16,4 @@ run:
@echo "Diffusers run."
test:
bash test.sh
bash test.sh