Replaced deepseek with qwq
This commit is contained in:
@@ -3,12 +3,12 @@ import asyncio
|
||||
|
||||
from .auto_approver import AutoApprover
|
||||
from .config import Config
|
||||
from .llm_engine.hf_llm_engine import HfLlmEngine
|
||||
from .llm_engine.deepseek_llm_engine import DeepSeekLlmEngine
|
||||
from .iteration_logger import IterationLogger
|
||||
from .llm_engine.hf_llm_engine import HfLlmEngine
|
||||
from .llm_engine.local_llm_engine import LocalLlmEngine
|
||||
from .llm_engine.mistral_llm_engine import MistralLlmEngine
|
||||
from .llm_engine.openai_llm_engine import OpenAILlmEngine
|
||||
from .llm_engine.qwq_llm_engine import QwQLlmEngine
|
||||
from .response_parser import ResponseParser
|
||||
from .system_metrics import SystemMetrics
|
||||
from .web.api import Api
|
||||
@@ -62,11 +62,11 @@ class Main:
|
||||
config.mistral_api_key,
|
||||
)
|
||||
|
||||
if config.deepseek_enabled:
|
||||
self._llms['deepseek'] = DeepSeekLlmEngine(
|
||||
config.deepseek_model,
|
||||
config.deepseek_temperature,
|
||||
config.deepseek_token_limit,
|
||||
if config.qwq_enabled:
|
||||
self._llms['qwq'] = QwQLlmEngine(
|
||||
config.qwq_model,
|
||||
config.qwq_temperature,
|
||||
config.qwq_token_limit,
|
||||
config.hf_api_key,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user