mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2025-05-28 03:10:28 -04:00
fix: Add hack for codellama fim models
This commit is contained in:
parent
5b9c67c2d8
commit
e90933d713
@ -30,7 +30,7 @@ class OllamaFim : public LLMCore::PromptTemplate
|
|||||||
public:
|
public:
|
||||||
LLMCore::TemplateType type() const override { return LLMCore::TemplateType::FIM; }
|
LLMCore::TemplateType type() const override { return LLMCore::TemplateType::FIM; }
|
||||||
QString name() const override { return "Ollama FIM"; }
|
QString name() const override { return "Ollama FIM"; }
|
||||||
QStringList stopWords() const override { return QStringList(); }
|
QStringList stopWords() const override { return QStringList() << "<EOT>"; }
|
||||||
void prepareRequest(QJsonObject &request, const LLMCore::ContextData &context) const override
|
void prepareRequest(QJsonObject &request, const LLMCore::ContextData &context) const override
|
||||||
{
|
{
|
||||||
request["prompt"] = context.prefix.value_or("");
|
request["prompt"] = context.prefix.value_or("");
|
||||||
|
Loading…
Reference in New Issue
Block a user