bot
Browse files
config.py
CHANGED
|
@@ -68,8 +68,8 @@ MAX_RECENT_HISTORY_MESSAGES = int(os.getenv("MAX_RECENT_HISTORY_MESSAGES", 6))
|
|
| 68 |
# "local" -> Mistral-7B-Instruct-v0.2 via transformers/bitsandbytes
|
| 69 |
# "groq" -> future Groq-hosted provider
|
| 70 |
# "openai"-> future OpenAI provider
|
| 71 |
-
|
| 72 |
-
LLM_PROVIDER = os.getenv("LLM_PROVIDER", "groq")
|
| 73 |
|
| 74 |
LOCAL_MODEL_NAME = os.getenv("LOCAL_MODEL_NAME", "mistralai/Mistral-7B-Instruct-v0.2")
|
| 75 |
LOCAL_MODEL_MAX_NEW_TOKENS = int(os.getenv("LOCAL_MODEL_MAX_NEW_TOKENS", 512))
|
|
|
|
| 68 |
# "local" -> Mistral-7B-Instruct-v0.2 via transformers/bitsandbytes
|
| 69 |
# "groq" -> future Groq-hosted provider
|
| 70 |
# "openai"-> future OpenAI provider
|
| 71 |
+
LLM_PROVIDER = os.getenv("LLM_PROVIDER", "local")
|
| 72 |
+
# LLM_PROVIDER = os.getenv("LLM_PROVIDER", "groq")
|
| 73 |
|
| 74 |
LOCAL_MODEL_NAME = os.getenv("LOCAL_MODEL_NAME", "mistralai/Mistral-7B-Instruct-v0.2")
|
| 75 |
LOCAL_MODEL_MAX_NEW_TOKENS = int(os.getenv("LOCAL_MODEL_MAX_NEW_TOKENS", 512))
|