mirror/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-04-16 21:08:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

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
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(cfg.Backend).To(Equal("transformers"))
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())
})
@@ -100,7 +100,7 @@ var _ = Describe("ImportLocalPath", func() {
cfg, err := importers.ImportLocalPath(modelDir, "merged")
Expect(err).ToNot(HaveOccurred())
Expect(cfg.Backend).To(Equal("transformers"))
Expect(cfg.Model).To(Equal(modelDir))
Expect(cfg.Model).To(Equal("merged"))
Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue())
})
@@ -113,7 +113,7 @@ var _ = Describe("ImportLocalPath", func() {
cfg, err := importers.ImportLocalPath(modelDir, "merged-pt")
Expect(err).ToNot(HaveOccurred())
Expect(cfg.Backend).To(Equal("transformers"))
Expect(cfg.Model).To(Equal(modelDir))
Expect(cfg.Model).To(Equal("merged-pt"))
})
})