How to use from the
Use from the
Transformers library
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("blenderwang/mol_id", dtype="auto")
Quick Links

Mol ID

A transformer encoder model pretrained on 50M ZINC SMILES string using flash attention 2

Hardware:

  • gpu that support flash attention 2 and bf16

Software:

  • flash attention 2
  • lightning for mixed precision (bf16-mixed)
  • wandb for logging
  • huggingface
    • tokenizers
    • datasets

github repo: link

Downloads last month
6
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support