tattabio/gLM2_650M_bgc_decoder

gLM2 650M BGC Decoder

A 650M-parameter masked discrete diffusion model for conditional generation and redesign of biosynthetic gene clusters. The model supports antiSMASH domain conditioning and sequences up to 16,384 tokens.

import torch
from transformers import AutoModelForMaskedLM, AutoTokenizer

model_id = "tattabio/gLM2_650M_bgc_decoder"

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForMaskedLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
    trust_remote_code=True,
).cuda()

Inference code: https://github.com/TattaBio/gLM2_decoder

Citation

TODO

License

Model weights are released under the Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0) license. The weights are freely available for academic and research purposes.