Fix safetensors size validation bug

quantizer.py

@@ -51,15 +51,20 @@ async def quantize_model(job: Dict) -> Dict:
 
             # Check size
             if hasattr(model_info, 'safetensors') and model_info.safetensors:
-                total_size = sum(
-                    file.size for file in model_info.safetensors.values()
-                )
-                size_gb = total_size / (1024**3)
-                print(f"  Model size: {size_gb:.2f} GB")
-
-                # Skip if too large (>10GB on free tier)
-                if size_gb > 10:
-                    raise Exception(f"Model too large for free tier: {size_gb:.2f} GB (max 10GB)")
+                total_size = 0
+                for file_info in model_info.safetensors.values():
+                    if isinstance(file_info, dict) and 'size' in file_info:
+                        total_size += file_info['size']
+                    elif hasattr(file_info, 'size'):
+                        total_size += file_info.size
+
+                if total_size > 0:
+                    size_gb = total_size / (1024**3)
+                    print(f"  Model size: {size_gb:.2f} GB")
+
+                    # Skip if too large (>10GB on free tier)
+                    if size_gb > 10:
+                        raise Exception(f"Model too large for free tier: {size_gb:.2f} GB (max 10GB)")
 
         except Exception as e:
             raise Exception(f"Model validation failed: {str(e)}")