This commit is contained in:
Isadora White 2025-03-22 11:34:57 -05:00
parent c9b52ffa5b
commit 8af595ffa8

View file

@ -24,11 +24,10 @@ export class VLLM {
async sendRequest(turns, systemMessage, stop_seq = '***') { async sendRequest(turns, systemMessage, stop_seq = '***') {
let messages = [{ 'role': 'system', 'content': systemMessage }].concat(turns); let messages = [{ 'role': 'system', 'content': systemMessage }].concat(turns);
if (this.model_name.includes("deepseek") || this.model_name.inclues("qwen")) { if (this.model_name.includes('deepseek') || this.model_name.includes('qwen')) {
messages = strictFormat(messages); messages = strictFormat(messages);
} }
const pack = { const pack = {
model: this.model_name || "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", model: this.model_name || "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
messages, messages,