akhaliq HF Staff commited on
Commit
232d5c7
·
1 Parent(s): 564ef48
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -2290,7 +2290,7 @@ AVAILABLE_MODELS = [
2290
  },
2291
  {
2292
  "name": "Kimi K2 Thinking",
2293
- "id": "kimi-k2-thinking",
2294
  "description": "Moonshot Kimi K2 Thinking model for advanced reasoning and code generation"
2295
  }
2296
  ]
@@ -2427,7 +2427,7 @@ def get_inference_client(model_id, provider="auto"):
2427
  api_key=os.getenv("MOONSHOT_API_KEY"),
2428
  base_url="https://api.moonshot.ai/v1",
2429
  )
2430
- elif model_id == "kimi-k2-thinking":
2431
  # Use HuggingFace InferenceClient with Novita provider for Kimi K2 Thinking
2432
  provider = "novita"
2433
  elif model_id == "stealth-model-1":
@@ -2472,9 +2472,6 @@ def get_real_model_id(model_id: str) -> str:
2472
  raise ValueError("STEALTH_MODEL_1_ID environment variable is required for Carrot model")
2473
 
2474
  return real_model_id
2475
- elif model_id == "kimi-k2-thinking":
2476
- # Return the full model name with provider tag for Kimi K2 Thinking
2477
- return "moonshotai/Kimi-K2-Thinking:novita"
2478
  return model_id
2479
 
2480
  # Type definitions
 
2290
  },
2291
  {
2292
  "name": "Kimi K2 Thinking",
2293
+ "id": "moonshotai/Kimi-K2-Thinking:novita",
2294
  "description": "Moonshot Kimi K2 Thinking model for advanced reasoning and code generation"
2295
  }
2296
  ]
 
2427
  api_key=os.getenv("MOONSHOT_API_KEY"),
2428
  base_url="https://api.moonshot.ai/v1",
2429
  )
2430
+ elif model_id == "moonshotai/Kimi-K2-Thinking:novita":
2431
  # Use HuggingFace InferenceClient with Novita provider for Kimi K2 Thinking
2432
  provider = "novita"
2433
  elif model_id == "stealth-model-1":
 
2472
  raise ValueError("STEALTH_MODEL_1_ID environment variable is required for Carrot model")
2473
 
2474
  return real_model_id
 
 
 
2475
  return model_id
2476
 
2477
  # Type definitions