File size: 265 Bytes
0646ed9 | 1 2 3 4 5 6 7 8 | """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)
|