Update rank_gpt.py

sunnweiwei · Jan 20, 2024 · 44167f8 · 44167f8
1 parent b925e41
commit 44167f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rank_gpt.py b/rank_gpt.py
@@ -177,8 +177,8 @@ def create_permutation_instruction(item=None, rank_start=0, rank_end=100, model_
 def run_llm(messages, api_key=None, model_name="gpt-3.5-turbo"):
     if 'gpt' in model_name:
         Client = OpenaiClient
-    # elif 'claude' in model_name:
-        # Client = ClaudeClient
+    elif 'claude' in model_name:
+        Client = ClaudeClient
     else:
         Client = LitellmClient