license: apache-2.0 | |
library_name: transformers | |
pipeline_tag: feature-extraction | |
tags: | |
- chemistry | |
- transformers | |
# selfies-ted2m | |
selfies-ted is an transformer based encoder decoder model for molecular representations using SELFIES. This is a 2.2M parameter version of the model. For the full-sized version and more information on architecture, see [selfies-ted](https://huggingface.co/ibm-research/materials.selfies-ted). | |
This version also includes a projection layer to convert the last hidden state of the BART model (256-dimensional vector per token) to a single 128-dimension vector for the whole SELFIES sequence. | |