Skip to content

Commit 8997ff6

Browse files
committed
Fix tests
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
1 parent f1223b4 commit 8997ff6

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

core/gallery/importers/local_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ var _ = Describe("ImportLocalPath", func() {
6868
Expect(err).ToNot(HaveOccurred())
6969
Expect(cfg.Backend).To(Equal("transformers"))
7070
Expect(cfg.Model).To(Equal("meta-llama/Llama-2-7b-hf"))
71-
Expect(cfg.LLMConfig.LoraAdapter).To(Equal(modelDir))
71+
Expect(cfg.LLMConfig.LoraAdapter).To(Equal("lora-model"))
7272
Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue())
7373
})
7474

@@ -100,7 +100,7 @@ var _ = Describe("ImportLocalPath", func() {
100100
cfg, err := importers.ImportLocalPath(modelDir, "merged")
101101
Expect(err).ToNot(HaveOccurred())
102102
Expect(cfg.Backend).To(Equal("transformers"))
103-
Expect(cfg.Model).To(Equal(modelDir))
103+
Expect(cfg.Model).To(Equal("merged"))
104104
Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue())
105105
})
106106

@@ -113,7 +113,7 @@ var _ = Describe("ImportLocalPath", func() {
113113
cfg, err := importers.ImportLocalPath(modelDir, "merged-pt")
114114
Expect(err).ToNot(HaveOccurred())
115115
Expect(cfg.Backend).To(Equal("transformers"))
116-
Expect(cfg.Model).To(Equal(modelDir))
116+
Expect(cfg.Model).To(Equal("merged-pt"))
117117
})
118118
})
119119

0 commit comments

Comments
 (0)