| """Install the bundled wheel, then load the public A3 encoder.""" | |
| from grasp import GRASPEncoder | |
| encoder=GRASPEncoder.from_pretrained("caithmac/GRASP-A3") | |
| vectors=encoder.encode(["CCO","N[C@@H](C)C(=O)O"]) | |
| assert vectors.shape==(2,768) | |
| print(vectors.shape) | |