Method get_model_info

utils/weights_manager.py:149–197

获取模型详细信息 Args: model_name: 模型名称 Returns: 模型信息字典

(self, model_name: str)

        return validation_results
    
    def get_model_info(self, model_name: str) -> Dict[str, Any]:
        """
        获取模型详细信息
        
        Args:
            model_name: 模型名称
            
        Returns:
            模型信息字典
        """
        model_path = self.get_model_path(model_name, create_if_missing=False)
        if not model_path:
            return {'exists': False}
        
        path_obj = Path(model_path)
        info = {
            'exists': True,
            'path': str(path_obj),
            'is_directory': path_obj.is_dir(),
            'size_mb': 0,
            'files': []
        }
        
        try:
            if path_obj.is_dir():
                # 计算目录大小和文件列表
                total_size = 0
                files = []
                for file_path in path_obj.rglob('*'):
                    if file_path.is_file():
                        size = file_path.stat().st_size
                        total_size += size
                        files.append({
                            'name': file_path.name,
                            'relative_path': str(file_path.relative_to(path_obj)),
                            'size_mb': round(size / (1024 * 1024), 2)
                        })
                info['size_mb'] = round(total_size / (1024 * 1024), 2)
                info['files'] = files
            else:
                # 单个文件
                size = path_obj.stat().st_size
                info['size_mb'] = round(size / (1024 * 1024), 2)
                info['files'] = [{'name': path_obj.name, 'size_mb': info['size_mb']}]
        except Exception as e:
            logger.error(f"获取模型信息失败: {e}")
            info['error'] = str(e)
        
        return info


# 全局权重管理器实例