Update crs_arena/battle_manager.py

crs_arena/battle_manager.py

@@ -36,13 +36,13 @@ CONVERSATION_COUNTS.update(
     {
         "kbrd_redial": 7,
         "kbrd_opendialkg": 7,
-        "unicrs_redial": 4,
-        "unicrs_opendialkg": 3,
-        "barcor_redial": 4,
-        "barcor_opendialkg": 10,
-        "crbcrs_redial": 10,
-        "chatgpt_opendialkg": 11,
-        "chatgpt_redial": 16,
+        "unicrs_redial": 10,
+        "unicrs_opendialkg": 10,
+        "barcor_redial": 20,
+        "barcor_opendialkg": 20,
+        "crbcrs_redial": 20,
+        "chatgpt_opendialkg": 20,
+        "chatgpt_redial": 20,
     }
 )
 
@@ -105,6 +105,10 @@ def cache_fighters(n: Optional[int] = None) -> None:
     """
     logging.info(f"Caching {n} CRS fighters.")
     for i, (model_name, config_path) in enumerate(CRS_MODELS.items()):
-        get_crs_model(model_name, config_path)
+        try:
+            get_crs_model(model_name, config_path)
+        except Exception as e:
+            logging.error(f"Caching CRS fighters: {e}")
+            get_crs_model.clear(model_name, config_path)
         if n is not None and i == n:
             break