vector-institute
/

atomformer-base

Graph Machine Learning

feature-extraction

Model card Files Files and versions

akore commited on Jun 26, 2024

Commit

efb9355

·

verified ·

1 Parent(s): e28f6a5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ coords = torch.randn(1, 10, 3)
 attn_mask = torch.ones(1, 10)
 output = model(input_ids, coords=coords, attention_mask=attention_mask)
-output[0].shape #  (torch.Size([1, 10, 768])
 ```

 attn_mask = torch.ones(1, 10)
 output = model(input_ids, coords=coords, attention_mask=attention_mask)
+output.shape #  (torch.Size([1, 10, 768])
 ```