feat: automatically move protein embeddings input to the right dtype e333d41 verified macwiatrak commited on 10 days ago