2024年7月——使用ESM3(进化尺度的蛋白质语言模型)

需要的库:huggingface

path = Path(snapshot_download(repo_id="EvolutionaryScale/esm3-sm-open-v1"))

权重文件会下载到当前工作目录下的 .cache/huggingface/datasets/downloads 目录中。

安装:pip install esm

(esm2安装是 pip install fair-esm ,)

下面是使用说明,其中可以设置login(token="YOUR_TOKEN_HERE")为自己的huggingface令牌

from huggingface_hub import login
from esm.models.esm3 import ESM3
from esm.sdk.api import ESM3InferenceClient, ESMProtein, GenerationConfig

# This will prompt you to get an API key from huggingface hub, make one with
# "Read" or "Write" permission and copy it back here.
login()

# This will download the model weights and instantiate the model on your machine.
model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cuda") # or "cpu"

# Generate a completi

上一篇:Spring Boot应用的部署与扩展-大家好,我是免费搭建查券返利机器人省钱赚佣金就用微赚淘客系统3.0的小编,也是冬天不穿秋裤,天冷也要风度的程序猿!


下一篇:假设性文档嵌入 HyDE:大模型 + 对比学习,从关键词相似度搜索到语义搜索-提出背景