From f401181cb5991fcce697749705864d1f99f13138 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 30 May 2023 12:35:32 +0200 Subject: [PATCH] fix: switch back to upstream for rwkv bindings (#432) --- Makefile | 4 ++-- api/api_test.go | 11 ++++++++--- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/Makefile b/Makefile index 8a0f62a..2d040e3 100644 --- a/Makefile +++ b/Makefile @@ -7,8 +7,8 @@ GOLLAMA_VERSION?=4bd3910005a593a6db237bc82c506d6d9fb81b18 GPT4ALL_REPO?=https://github.com/nomic-ai/gpt4all GPT4ALL_VERSION?=73db20ba85fbbdc66a56e2619394c0eea40dc72b GOGGMLTRANSFORMERS_VERSION?=695f97befe14f0107d8da1c11f5b84912e0754b6 -RWKV_REPO?=https://github.com/mudler/go-rwkv.cpp -RWKV_VERSION?=dcbd34aff983b3d04fa300c5da5ec4bfdf6db295 +RWKV_REPO?=https://github.com/donomii/go-rwkv.cpp +RWKV_VERSION?=ccb05c3e1c6efd098017d114dcb58ab3262b40b2 WHISPER_CPP_VERSION?=9b926844e3ae0ca6a0d13573b2e0349be1a4b573 BERT_VERSION?=cea1ed76a7f48ef386a8e369f6c82c48cdf2d551 BLOOMZ_VERSION?=1834e77b83faafe912ad4092ccf7f77937349e2f diff --git a/api/api_test.go b/api/api_test.go index 4b24514..2fa7e43 100644 --- a/api/api_test.go +++ b/api/api_test.go @@ -114,7 +114,8 @@ var _ = Describe("API test", func() { modelLoader = model.NewModelLoader(tmpdir) c, cancel = context.WithCancel(context.Background()) - app = App(WithContext(c), WithModelLoader(modelLoader)) + app, err = App(WithContext(c), WithModelLoader(modelLoader)) + Expect(err).ToNot(HaveOccurred()) go app.Listen("127.0.0.1:9090") defaultConfig := openai.DefaultConfig("") @@ -198,7 +199,9 @@ var _ = Describe("API test", func() { modelLoader = model.NewModelLoader(os.Getenv("MODELS_PATH")) c, cancel = context.WithCancel(context.Background()) - app = App(WithContext(c), WithModelLoader(modelLoader)) + var err error + app, err = App(WithContext(c), WithModelLoader(modelLoader)) + Expect(err).ToNot(HaveOccurred()) go app.Listen("127.0.0.1:9090") defaultConfig := openai.DefaultConfig("") @@ -316,7 +319,9 @@ var _ = Describe("API test", func() { modelLoader = model.NewModelLoader(os.Getenv("MODELS_PATH")) c, cancel = context.WithCancel(context.Background()) - app = App(WithContext(c), WithModelLoader(modelLoader), WithConfigFile(os.Getenv("CONFIG_FILE"))) + var err error + app, err = App(WithContext(c), WithModelLoader(modelLoader), WithConfigFile(os.Getenv("CONFIG_FILE"))) + Expect(err).ToNot(HaveOccurred()) go app.Listen("127.0.0.1:9090") defaultConfig := openai.DefaultConfig("")