用途
本技能指导用户将 Profluent E1 蛋白质语言模型部署为 ToolUniverse 远程工具,提供 MCP 接口,并可选通过 ToolUniverse Connect 进行中继共享。适用于需要在本地 GPU 上运行 E1 模型进行蛋白质变体打分的场景。
主要能力
- 提供
run_e1_score_variants工具,对给定父序列和突变序列计算零样本掩码边际对数似然比。 - 提供
run_e1_embed工具,默认返回平均池化向量,并可将逐残基矩阵保存为 float32 .npy 文件路径。 - 支持通过
tu remote setup e1自动完成预检、环境创建、套件安装、权重下载和冒烟测试。 - 支持通过
tu remote share e1进行远程共享,使用浏览器设备授权。
使用方式
- 前置条件:Linux 系统、Python 3.12.3、CUDA GPU,从 ToolUniverse 仓库根目录运行。
- 运行契约检查:
python scripts/remote_validation/setup_skill_preflight.py --implementation e1。 - 创建两个虚拟环境:ToolUniverse 环境(用于 MCP 服务器)和模型环境(由
TOOLUNIVERSE_E1_PYTHON指定)。 - 设置
HF_HOME和TOOLUNIVERSE_E1_PYTHON,然后运行tu remote setup e1。可加--only preflight|environment|kit|weights|smoke分步执行。 - 本地启动:
python -m tooluniverse.remote.e1.e1_tool,端点默认http://127.0.0.1:8044/mcp。 - 验证工具发现:使用 FastMCP 客户端列出工具,确认包含
run_e1_score_variants。 - 远程共享:安装
tuplatform-connectwheel,运行tu remote login后执行tu remote share e1。
适用场景
- 蛋白质工程中的变体效应预测。
- 需要本地化、私有部署的蛋白质语言模型服务。
- 通过 MCP 协议将 E1 集成到智能体工作流。
限制
- 仅支持 Linux + CUDA GPU,模型常驻显存,CPU 不是目标。
- 当前验证状态为私有部署,公开分享、跨用户隔离、并发、恢复和评分质量未测试。
- 需保持绑定回环地址,非回环绑定需配置
TOOLUNIVERSE_API_TOKEN。 - 权重和缓存不应纳入 Git。
- 首次请求有模型构建延迟(约 12 秒),后续请求约 0.075-0.080 秒。
- 峰值显存随输入长度增长,100 残基约 4.30 GiB,800 残基约 4.84 GiB。
- 仅使用实现 README 链接的官方上游文档,不得使用第三方模型镜像。