Update leaderboard.py

leaderboard.py

@@ -31,11 +31,50 @@ def save_leaderboard(leaderboard_data: Dict[str, Any]) -> bool:
         return False
 
 def get_model_size(model_name):
+    """Extract model size in billions from model name.
+    Handles various formats like:
+    - "Model 14B (4-bit)"
+    - "Model (14B)"
+    - "Model 14.5B"
+    - "Model 1,000M"
+    """
     for model, human_readable in config.get_approved_models():
         if model == model_name:
-            size = float(human_readable.split('(')[1].split('B')[0])
-            return size
-    return 1.0  # Default size if not found
+            try:
+                # Remove any commas
+                clean_name = human_readable.replace(',', '')
+                
+                # Try to find size in parentheses first
+                if '(' in clean_name:
+                    parts = clean_name.split('(')
+                    for part in parts:
+                        if 'B' in part:
+                            size_str = part.split('B')[0].strip()
+                            try:
+                                return float(size_str)
+                            except ValueError:
+                                continue
+                
+                # If not in parentheses, look for B or M in the whole string
+                words = clean_name.split()
+                for word in words:
+                    if 'B' in word:
+                        size_str = word.replace('B', '').strip()
+                        try:
+                            return float(size_str)
+                        except ValueError:
+                            continue
+                    elif 'M' in word:
+                        size_str = word.replace('M', '').strip()
+                        try:
+                            return float(size_str) / 1000  # Convert millions to billions
+                        except ValueError:
+                            continue
+                            
+            except Exception as e:
+                print(f"Error parsing size for {model_name}: {e}")
+                
+    return 1.0  # Default size if not found or parsing failed
 
 def calculate_expected_score(rating_a, rating_b):
     return 1 / (1 + math.pow(10, (rating_b - rating_a) / 400))