File size: 1,265 Bytes
f4c651c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | ---
language: en
library_name: transformers
pipeline_tag: text-generation
tags:
- t5
- molecule-to-protein
- smiles
- protein-generation
- binder
- ligand
license: apache-2.0
datasets:
- contributor-anonymous/Mol2Pro-Binder-Dataset
---
# Mol2Pro-base
## Model description
- **Architecture:** T5-efficient-base https://huggingface.co/google/t5-efficient-base
- **Tokenization:** https://huggingface.co/contributor-anonymous/Mol2Pro-tokenizer
- **Code:** https://github.com/contributor-anonymous/Mol2Pro-tools
- **Training data** https://huggingface.co/datasets/contributor-anonymous/Mol2Pro-Binder-Dataset
## How to use
```python
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import torch
model_id = "contributor-anonymous/Mol2Pro-base"
tokenizer_id = "contributor-anonymous/Mol2Pro-tokenizer"
# Load tokenizers
tokenizer_mol = AutoTokenizer.from_pretrained(tokenizer_id, subfolder="smiles")
tokenizer_aa = AutoTokenizer.from_pretrained(tokenizer_id, subfolder="aa")
# Load model
model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
```
## Intended use
Research use only. The model generates candidate sequences conditioned on small-molecule inputs; it does not guarantee binding or function and must be validated experimentally.
|