From ec2adc2c031e106a23bc1438bb3f8d6517f22ff8 Mon Sep 17 00:00:00 2001 From: mudler Date: Fri, 5 May 2023 22:07:01 +0200 Subject: [PATCH] tests: use 3 cores --- api/api_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/api_test.go b/api/api_test.go index 9682a21..6f21f21 100644 --- a/api/api_test.go +++ b/api/api_test.go @@ -23,7 +23,7 @@ var _ = Describe("API test", func() { Context("API query", func() { BeforeEach(func() { modelLoader = model.NewModelLoader(os.Getenv("MODELS_PATH")) - app = App("", modelLoader, 1, 512, false, true, true) + app = App("", modelLoader, 3, 512, false, true, true) go app.Listen("127.0.0.1:9090") defaultConfig := openai.DefaultConfig("") @@ -87,7 +87,7 @@ var _ = Describe("API test", func() { Context("Config file", func() { BeforeEach(func() { modelLoader = model.NewModelLoader(os.Getenv("MODELS_PATH")) - app = App(os.Getenv("CONFIG_FILE"), modelLoader, 1, 512, false, true, true) + app = App(os.Getenv("CONFIG_FILE"), modelLoader, 3, 512, false, true, true) go app.Listen("127.0.0.1:9090") defaultConfig := openai.DefaultConfig("")