@@ -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