forge-api

# Forge API 参考 ## 概述 Forge 是 EvolutionaryScale 推出的云平台，用于可扩展的蛋白质设计和推理。它提供对完整 ESM3 模型家族的 API 访问，包括无法在本地运行的大型模型。 **核心优势：** - 访问所有 ESM3 模型，包括 98B 参数版本 - 无需本地 GPU - 可扩展的批量处理 - 自动更新至最新模型 - 生产级基础设施 - 支持异步/并发请求 ## 快速入门 ### 1. 获取 API Token 注册并获取您的 API Token：https://forge.evolutionaryscale.ai ### 2. 安装 ESM SDK bash pip install esm Forge 客户端包含在标准的 ESM 包中。 ### 3. 基础连接 python from esm.sdk.forge import ESM3ForgeInferenceClient from esm.sdk.api import ESMProtein, GenerationConfig # 初始化客户端 client = ESM3ForgeInferenceClient( model="esm3-medium-2024-08", url="https://forge.evolutionaryscale.ai", token="" ) # 测试连接 protein = ESMProtein(sequence="MPRT___KEND") result = client.generate(protein, GenerationConfig(track="sequence", num_steps=8)) print(result.sequence) ## 可用模型 | 模型 ID | 参数量 | 速度 | 质量 | 适用场景 | |----------|-----------|-------|---------|----------| | `esm3-small-2024-08` | 1.4B | 最快 | 良好 | 快速原型、测试 | | `esm3-medium-2024-08` | 7B | 快 | 优秀 | 生产环境、大多数应用 | | `esm3-large-2024-03` | 98B | 较慢 | 最佳 | 科研、关键设计 | | `esm3-medium-multimer-2024-09` | 7B | 快 | 实验性 | 蛋白质复合物 | **模型选择指南：** - **开发/测试**: 使用 `esm3-small-2024-08` 进行快速迭代 - **生产环境**: 使用 `esm3-medium-2024-08` 以获得最佳平衡 - **科研/关键任务**: 使用 `esm3-large-2024-03` 以获得最高质量 - **复合物**: 使用 `esm3-medium-multimer-2024-09` (实验性) ## ESM3ForgeInferenceClient API ### 初始化 python from esm.sdk.forge import ESM3ForgeInferenceClient # 基础初始化 client = ESM3ForgeInferenceClient( model="esm3-medium-2024-08", token="" ) # 使用自定义 URL (用于企业部署) client = ESM3ForgeInferenceClient( model="esm3-medium-2024-08", url="https://custom.forge.instance.com", token="" ) # 使用超时配置 client = ESM3ForgeInferenceClient( model="esm3-medium-2024-08", token="", timeout=300