added

Author: efrick2002

fastchat/serve/gradio_block_arena_anony.py

@@ -175,25 +175,7 @@ def share_click(state0, state1, model_selector0, model_selector1, request: gr.Re
         )
 
 
-SAMPLING_WEIGHTS = {'claude-3-5-haiku-20241022': 1,
- 'claude-3-5-sonnet-20240620': 1,
- 'claude-3-5-sonnet-20241022': 1,
- 'gpt-4-1106-preview': 1,
- 'gpt-4-turbo-2024-04-09': 1,
- 'gpt-4o-2024-05-13': 1,
- 'gpt-4o-2024-08-06': 1,
- 'gpt-4o-mini-2024-07-18': 1,
- 'o1-mini': 1,
- 'yi-lightning': 1,
- 'llama-3.1-405b-instruct-fp8': 1,
- 'llama-3.1-8b-instruct': 1,
- 'llama-3.1-70b-instruct': 1,
- 'gemini-1.5-pro-001': 1,
- 'mistral-large-2407': 1,
- 'qwen2.5-72b-instruct': 1,
- 'gemma-2-27b-it': 1,
- 'glm-4-plus': 1,
- 'p2l': 1000}
+SAMPLING_WEIGHTS = {}
 
 # target model sampling weights will be boosted.
 BATTLE_TARGETS = {}

pyproject.toml

@@ -19,7 +19,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-model_worker = ["accelerate>=0.21", "peft", "sentencepiece", "torch", "transformers>=4.31.0", "protobuf", "openai"]
+model_worker = ["accelerate>=0.21", "peft", "sentencepiece", "torch", "transformers>=4.31.0", "protobuf", "openai", "anthropic"]
 webui = ["gradio>=4.10", "plotly", "scipy"]
 train = ["einops", "flash-attn>=2.0", "wandb"]
 llm_judge = ["openai<1", "anthropic>=0.3", "ray"]