bep40 commited on
Commit
520acbc
·
verified ·
1 Parent(s): 1fd2dc2

Fix title generation: add huggingface/ prefix for litellm

Browse files
Files changed (1) hide show
  1. patch_models.py +2 -2
patch_models.py CHANGED
@@ -27,9 +27,9 @@ if target in content:
27
  content = content.replace(target, target + patch)
28
  print("✅ Added models to AVAILABLE_MODELS")
29
 
30
- # Fix title generation
31
  old_title = 'model="openai/openai/gpt-oss-120b:cerebras",\n api_base="https://router.huggingface.co/v1",\n api_key=api_key,'
32
- new_title = 'model="deepseek-ai/DeepSeek-V4-Pro",\n api_base=_os.environ.get("OPENAI_API_BASE", "https://router.huggingface.co/v1"),\n api_key=_os.environ.get("OPENAI_API_KEY", api_key),'
33
  if old_title in content:
34
  content = content.replace(old_title, new_title)
35
  content = content.replace(' reasoning_effort="low",\n', '')
 
27
  content = content.replace(target, target + patch)
28
  print("✅ Added models to AVAILABLE_MODELS")
29
 
30
+ # Fix title generation - use huggingface/ prefix for litellm
31
  old_title = 'model="openai/openai/gpt-oss-120b:cerebras",\n api_base="https://router.huggingface.co/v1",\n api_key=api_key,'
32
+ new_title = 'model="huggingface/deepseek-ai/DeepSeek-V4-Pro",\n api_key=api_key,'
33
  if old_title in content:
34
  content = content.replace(old_title, new_title)
35
  content = content.replace(' reasoning_effort="low",\n', '')