Fix ip_address bug

THUDM · Oct 31, 2023 · e33a45d · e33a45d
1 parent 42c9dd2
commit e33a45d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval_general/eval_gsm8k_tgi.py b/eval_general/eval_gsm8k_tgi.py
@@ -106,7 +106,7 @@ def clean_answer(text):
 
 def generate_sample(prompt, ip, post_from, post_to):
     ip_address = f"http://{ip}:"
-    controller_addr_pool = [ip + str(i) for i in range(post_from, post_to + 1)]
+    controller_addr_pool = [ip_address + str(i) for i in range(post_from, post_to + 1)]
 
     def request_answer(max_new_tokens):
         data = {