小分子的语言模型MolT5的使用

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

前言

这是MolT5的安装和embedding获取（gpu版）。

零、安装

conda activate base conda install-c conda-forge mamba-y conda create-n molt5 python=3.9-y conda activate molt5 mamba install pytorch pytorch-cuda=11.8-c pytorch-c nvidia-y pip install sentencepiece accelerate mamba install-c conda-forge rdkit-y pip install"transformers==4.38.2"# 需要离线的，去下载https://huggingface.co/laituan245/molt5-base，作为./molt5-base

一、使用步骤

1.引入库

import torch from transformers import T5Tokenizer,T5EncoderModel

2.获取embedding

device="cuda"