File size: 637 Bytes
d202de9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
---
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.