Zenaight commited on
Commit
40c7bdf
·
verified ·
1 Parent(s): fae2c16

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +2 -1
app/main.py CHANGED
@@ -115,6 +115,7 @@ class HFInferenceLLM(LLM):
115
  provider: str = "hf-inference"
116
  api_key: str = ""
117
  max_tokens: int = 4000 # Increased from 500 to 4000 by default
 
118
 
119
  def __init__(self, model: str, provider: str = "hf-inference", api_key: str = "", max_tokens: int = 4000):
120
  super().__init__()
@@ -216,7 +217,7 @@ def get_llm(model_name: str):
216
  try:
217
  return HFInferenceLLM(
218
  model="mistralai/Mistral-7B-Instruct-v0.3",
219
- provider="hf-inference",
220
  api_key=hf_token,
221
  max_tokens=4000
222
  )
 
115
  provider: str = "hf-inference"
116
  api_key: str = ""
117
  max_tokens: int = 4000 # Increased from 500 to 4000 by default
118
+
119
 
120
  def __init__(self, model: str, provider: str = "hf-inference", api_key: str = "", max_tokens: int = 4000):
121
  super().__init__()
 
217
  try:
218
  return HFInferenceLLM(
219
  model="mistralai/Mistral-7B-Instruct-v0.3",
220
+ provider="together",
221
  api_key=hf_token,
222
  max_tokens=4000
223
  )