diff --git a/Dockerfile b/Dockerfile index bb07b53..db31e01 100644 --- a/Dockerfile +++ b/Dockerfile @@ -53,10 +53,13 @@ RUN curl -L "https://github.com/gabime/spdlog/archive/refs/tags/v${SPDLOG_VERSIO tar -C "lib/Linux-$(uname -m)/piper_phonemize" -xzvf - && ls -liah /build/lib/Linux-$(uname -m)/piper_phonemize/ && \ cp -rfv /build/lib/Linux-$(uname -m)/piper_phonemize/lib/. /lib64/ && \ cp -rfv /build/lib/Linux-$(uname -m)/piper_phonemize/lib/. /usr/lib/ && \ - cp -rfv /build/lib/Linux-$(uname -m)/piper_phonemize/include/. /usr/include/ + cp -rfv /build/lib/Linux-$(uname -m)/piper_phonemize/include/. /usr/include/ # \ # ; fi +################################### +################################### + FROM requirements as builder ARG GO_TAGS="stablediffusion tts" @@ -66,9 +69,18 @@ ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility ENV NVIDIA_REQUIRE_CUDA="cuda>=${CUDA_MAJOR_VERSION}.0" ENV NVIDIA_VISIBLE_DEVICES=all +WORKDIR /build + +COPY Makefile . +RUN make get-sources +COPY go.mod . +RUN make prepare COPY . . RUN ESPEAK_DATA=/build/lib/Linux-$(uname -m)/piper_phonemize/lib/espeak-ng-data make build +################################### +################################### + FROM requirements ARG FFMPEG @@ -83,6 +95,10 @@ RUN if [ "${FFMPEG}" = "true" ]; then \ WORKDIR /build +# we start fresh & re-copy all assets because `make build` does not clean up nicely after itself +# so when `entrypoint.sh` runs `make build` again (which it does by default), the build would fail +# see https://github.com/go-skynet/LocalAI/pull/658#discussion_r1241971626 and +# https://github.com/go-skynet/LocalAI/pull/434 COPY . . RUN make prepare-sources COPY --from=builder /build/local-ai ./ @@ -92,4 +108,4 @@ HEALTHCHECK --interval=1m --timeout=10m --retries=10 \ CMD curl -f $HEALTHCHECK_ENDPOINT || exit 1 EXPOSE 8080 -ENTRYPOINT [ "/build/entrypoint.sh" ] \ No newline at end of file +ENTRYPOINT [ "/build/entrypoint.sh" ] diff --git a/Makefile b/Makefile index f14f3d4..4b8b149 100644 --- a/Makefile +++ b/Makefile @@ -162,7 +162,7 @@ gpt4all/gpt4all-bindings/golang/libgpt4all.a: gpt4all $(MAKE) -C gpt4all/gpt4all-bindings/golang/ libgpt4all.a ## CEREBRAS GPT -go-ggml-transformers: +go-ggml-transformers: git clone --recurse-submodules https://github.com/go-skynet/go-ggml-transformers.cpp go-ggml-transformers cd go-ggml-transformers && git checkout -b build $(GOGPT2_VERSION) && git submodule update --init --recursive --depth 1 # This is hackish, but needed as both go-llama and go-gpt4allj have their own version of ggml.. @@ -194,12 +194,15 @@ go-llama: git clone --recurse-submodules https://github.com/go-skynet/go-llama.cpp go-llama cd go-llama && git checkout -b build $(GOLLAMA_VERSION) && git submodule update --init --recursive --depth 1 -go-llama/libbinding.a: go-llama +go-llama/libbinding.a: go-llama $(MAKE) -C go-llama BUILD_TYPE=$(BUILD_TYPE) libbinding.a go-piper/libpiper_binding.a: $(MAKE) -C go-piper libpiper_binding.a example/main +get-sources: go-llama go-ggml-transformers gpt4all go-piper go-rwkv whisper.cpp go-bert bloomz go-stable-diffusion + touch $@ + replace: $(GOCMD) mod edit -replace github.com/go-skynet/go-llama.cpp=$(shell pwd)/go-llama $(GOCMD) mod edit -replace github.com/nomic-ai/gpt4all/gpt4all-bindings/golang=$(shell pwd)/gpt4all/gpt4all-bindings/golang @@ -211,7 +214,7 @@ replace: $(GOCMD) mod edit -replace github.com/mudler/go-stable-diffusion=$(shell pwd)/go-stable-diffusion $(GOCMD) mod edit -replace github.com/mudler/go-piper=$(shell pwd)/go-piper -prepare-sources: go-llama go-ggml-transformers gpt4all go-piper go-rwkv whisper.cpp go-bert bloomz go-stable-diffusion replace +prepare-sources: get-sources replace $(GOCMD) mod download ## GENERIC @@ -228,6 +231,7 @@ rebuild: ## Rebuilds the project $(MAKE) build prepare: prepare-sources backend-assets/gpt4all $(OPTIONAL_TARGETS) go-llama/libbinding.a go-bert/libgobert.a go-ggml-transformers/libtransformers.a go-rwkv/librwkv.a whisper.cpp/libwhisper.a bloomz/libbloomz.a ## Prepares for building + touch $@ clean: ## Remove build related file rm -fr ./go-llama