Spaces:
Running
Running
update minimax to use inference providers
Browse files
app.py
CHANGED
|
@@ -2428,8 +2428,8 @@ AVAILABLE_MODELS = [
|
|
| 2428 |
},
|
| 2429 |
{
|
| 2430 |
"name": "MiniMax M2",
|
| 2431 |
-
"id": "MiniMax-M2",
|
| 2432 |
-
"description": "MiniMax M2 model via
|
| 2433 |
}
|
| 2434 |
]
|
| 2435 |
# Default model selection
|
|
@@ -2522,12 +2522,9 @@ def get_inference_client(model_id, provider="auto"):
|
|
| 2522 |
api_key=os.getenv("OPENROUTER_API_KEY"),
|
| 2523 |
base_url="https://openrouter.ai/api/v1",
|
| 2524 |
)
|
| 2525 |
-
elif model_id == "MiniMax-M2":
|
| 2526 |
-
# Use
|
| 2527 |
-
|
| 2528 |
-
api_key=os.getenv("MINIMAX_API_KEY"),
|
| 2529 |
-
base_url="https://api.minimax.io/v1",
|
| 2530 |
-
)
|
| 2531 |
elif model_id == "step-3":
|
| 2532 |
# Use StepFun API client for Step-3 model
|
| 2533 |
return OpenAI(
|
|
|
|
| 2428 |
},
|
| 2429 |
{
|
| 2430 |
"name": "MiniMax M2",
|
| 2431 |
+
"id": "MiniMaxAI/MiniMax-M2",
|
| 2432 |
+
"description": "MiniMax M2 model via HuggingFace InferenceClient with Novita provider"
|
| 2433 |
}
|
| 2434 |
]
|
| 2435 |
# Default model selection
|
|
|
|
| 2522 |
api_key=os.getenv("OPENROUTER_API_KEY"),
|
| 2523 |
base_url="https://openrouter.ai/api/v1",
|
| 2524 |
)
|
| 2525 |
+
elif model_id == "MiniMaxAI/MiniMax-M2":
|
| 2526 |
+
# Use HuggingFace InferenceClient with Novita provider for MiniMax M2 model
|
| 2527 |
+
provider = "novita"
|
|
|
|
|
|
|
|
|
|
| 2528 |
elif model_id == "step-3":
|
| 2529 |
# Use StepFun API client for Step-3 model
|
| 2530 |
return OpenAI(
|