Spaces:
Runtime error
Runtime error
Update app/main.py
Browse files- 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="
|
| 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 |
)
|