From 1c64c83909e74f44aae8a2d574d4d6d643ad041d Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 21 Jan 2025 16:36:15 +0100 Subject: [PATCH] chore(tests): try to fixup current tests Signed-off-by: Ettore Di Giacinto --- core/http/app_test.go | 3 ++- core/http/endpoints/openai/image.go | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/core/http/app_test.go b/core/http/app_test.go index 46833e10..454038ba 100644 --- a/core/http/app_test.go +++ b/core/http/app_test.go @@ -768,7 +768,8 @@ var _ = Describe("API test", func() { } response := postModelApplyRequest("http://127.0.0.1:9090/models/apply", modelApplyRequest{ - ID: "localai@sd-3.5-large-ggml", + ID: "localai@sd-3.5-large-ggml", + Name: "stablediffusion", }) Expect(response["uuid"]).ToNot(BeEmpty(), fmt.Sprint(response)) diff --git a/core/http/endpoints/openai/image.go b/core/http/endpoints/openai/image.go index de64d2d2..bd3f0987 100644 --- a/core/http/endpoints/openai/image.go +++ b/core/http/endpoints/openai/image.go @@ -72,7 +72,7 @@ func ImageEndpoint(cl *config.BackendConfigLoader, ml *model.ModelLoader, appCon } if m == "" { - m = model.StableDiffusionGGMLBackend + m = "stablediffusion" } log.Debug().Msgf("Loading model: %+v", m)