Fix providers and prompt default value

This commit is contained in:
Petr Mironychev
2024-10-02 22:25:53 +02:00
parent 1cbde3d55b
commit d235d0fcdf
4 changed files with 19 additions and 8 deletions

View File

@ -27,7 +27,7 @@ class CodeLlamaFimTemplate : public PromptTemplate
{
public:
TemplateType type() const override { return TemplateType::Fim; }
QString name() const override { return "CodeLlama FIM"; }
QString name() const override { return "CodeLLama FIM"; }
QString promptTemplate() const override { return "%1<PRE> %2 <SUF>%3 <MID>"; }
QStringList stopWords() const override
{

View File

@ -28,7 +28,7 @@ class CodeLlamaInstructTemplate : public PromptTemplate
{
public:
TemplateType type() const override { return TemplateType::Chat; }
QString name() const override { return "CodeLlama Chat"; }
QString name() const override { return "CodeLLama Chat"; }
QString promptTemplate() const override { return "[INST] %1 [/INST]"; }
QStringList stopWords() const override { return QStringList() << "[INST]" << "[/INST]"; }