Elgato_dark/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI synced 2026-04-21 13:27:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 29

Fix tests

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2026-03-21 01:04:28 +00:00
parent f1223b45b2
commit 8997ff6042
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
core/gallery/importers/local_test.go
View file

@ -68,7 +68,7 @@ var _ = Describe("ImportLocalPath", func() {
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(cfg.Backend).To(Equal("transformers")) Expect(cfg.Backend).To(Equal("transformers"))
Expect(cfg.Model).To(Equal("meta-llama/Llama-2-7b-hf")) Expect(cfg.Model).To(Equal("meta-llama/Llama-2-7b-hf"))
Expect(cfg.LLMConfig.LoraAdapter).To(Equal(modelDir)) Expect(cfg.LLMConfig.LoraAdapter).To(Equal("lora-model"))
Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue()) Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue())
}) })
@ -100,7 +100,7 @@ var _ = Describe("ImportLocalPath", func() {
cfg, err := importers.ImportLocalPath(modelDir, "merged") cfg, err := importers.ImportLocalPath(modelDir, "merged")
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(cfg.Backend).To(Equal("transformers")) Expect(cfg.Backend).To(Equal("transformers"))
Expect(cfg.Model).To(Equal(modelDir)) Expect(cfg.Model).To(Equal("merged"))
Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue()) Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue())
}) })
@ -113,7 +113,7 @@ var _ = Describe("ImportLocalPath", func() {
cfg, err := importers.ImportLocalPath(modelDir, "merged-pt") cfg, err := importers.ImportLocalPath(modelDir, "merged-pt")
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(cfg.Backend).To(Equal("transformers")) Expect(cfg.Backend).To(Equal("transformers"))
Expect(cfg.Model).To(Equal(modelDir)) Expect(cfg.Model).To(Equal("merged-pt"))
}) })
}) })