Add StarCoder2 instruct support

This commit is contained in:
Petr Mironychev
2024-10-16 10:45:48 +02:00
parent 9158a3ac0d
commit 46829720d8
4 changed files with 55 additions and 1 deletions

View File

@ -55,6 +55,7 @@
#include "templates/DeepSeekCoderFim.hpp"
#include "templates/QwenChat.hpp"
#include "templates/StarCoder2Fim.hpp"
#include "templates/StarCoderChat.hpp"
using namespace Utils;
using namespace Core;
@ -95,6 +96,7 @@ public:
templateManager.registerTemplate<Templates::CodeLlamaChat>();
templateManager.registerTemplate<Templates::QwenChat>();
templateManager.registerTemplate<Templates::LlamaChat>();
templateManager.registerTemplate<Templates::StarCoderChat>();
Utils::Icon QCODEASSIST_ICON(
{{":/resources/images/qoderassist-icon.png", Utils::Theme::IconsBaseColor}});