Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

除了stable-code之外支持其他的本地LLM吗 #37

Open
LoadingZhang opened this issue Jun 13, 2024 · 1 comment
Open

除了stable-code之外支持其他的本地LLM吗 #37

LoadingZhang opened this issue Jun 13, 2024 · 1 comment

Comments

@LoadingZhang
Copy link

LoadingZhang commented Jun 13, 2024

看了下代码只匹配了stable-code模型

@TonyJiangWJ
Copy link

支持的 就是需要自己修改代码 每一个模型的PROMPT写法不一样 需要按它的实际要求修改 可以参考 504行 就是stable-code要求的模板

override/main.go

Lines 501 to 504 in 8fdd840

func constructWithStableCodeModel(body []byte) []byte {
suffix := gjson.GetBytes(body, "suffix")
prompt := gjson.GetBytes(body, "prompt")
content := fmt.Sprintf("<fim_prefix>%s<fim_suffix>%s<fim_middle>", prompt, suffix)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants