diff --git a/vicuna-tools.d/main.go b/vicuna-tools.d/main.go index a5b00af..dc18be9 100644 --- a/vicuna-tools.d/main.go +++ b/vicuna-tools.d/main.go @@ -302,7 +302,7 @@ func handleAPIChatBotPut(w http.ResponseWriter, r *http.Request) error { if err := appendFile(inputF, reversePrompt+message); err != nil { return err } - if _, err := os.Stat(cacheF); false && os.IsNotExist(err) { + if _, err := os.Stat(cacheF); os.IsNotExist(err) { if err := func() error { commands := strings.Fields(Config.ChatBot.Command) commands = append(commands, @@ -333,8 +333,8 @@ func handleAPIChatBotPut(w http.ResponseWriter, r *http.Request) error { commands := strings.Fields(Config.ChatBot.Command) commands = append(commands, "-f", inputF, - //"--prompt-cache-all", - //"--prompt-cache", cacheF, + "--prompt-cache-all", + "--prompt-cache", cacheF, "-n", strconv.Itoa(Config.ChatBot.N), "--reverse-prompt", reversePrompt, )