akhaliq HF Staff commited on
Commit
a2250b5
·
1 Parent(s): 02ae3ab

update minimax to use inference providers

Browse files
Files changed (1) hide show
  1. app.py +5 -8
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 MiniMax API for code generation and general tasks"
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 MiniMax API client for MiniMax M2 model
2527
- return OpenAI(
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(