mirror/LocalAI
1
0 Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-09-25 07:34:58 -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 committed 2026-03-21 01:04:28 +00:00
1 parent f1223b45b2
commit 8997ff6042
1 file changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files
core/gallery/importers/local_test.go
+3 -3
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"))
})
})