check if hm in models before deleting to avoid errors (#2870)

Co-authored-by: Your Name <[email protected]>
lm-sys · Dec 30, 2023 · 722ab02 · 722ab02
1 parent 4735aa7
commit 722ab02
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py
@@ -155,7 +155,8 @@ def get_model_list(
 
     hidden_models = ["gpt-4-0314", "gpt-4-0613"]
     for hm in hidden_models:
-        del models[models.index(hm)]
+        if hm in models:
+            del models[models.index(hm)]
 
     priority = {k: f"___{i:03d}" for i, k in enumerate(model_info)}
     models.sort(key=lambda x: priority.get(x, x))