Skip to content

Ume model class and embeddings tutorial notebook#48

Merged
karinazad merged 4 commits intomainfrom
ume-notebook-embeddings
Mar 19, 2025
Merged

Ume model class and embeddings tutorial notebook#48
karinazad merged 4 commits intomainfrom
ume-notebook-embeddings

Conversation

@karinazad
Copy link
Collaborator

No description provided.


Parameters
----------
checkpoint : str
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can we make this accessible from local & HF?

>>> tokenizer = encoder.get_tokenizer(sequences, "amino_acid")
>>> tokens = tokenizer(sequences, return_tensors="pt")
"""
modality_enum = Modality(modality)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TODOs: this MR or future MR
can we add this as a @property to print the available modalities?

as well as convenience methods for saving the model and tokenizer locally and to HF, and naturalness (pseudo likelihood)

@karinazad karinazad merged commit 969afe2 into main Mar 19, 2025
5 checks passed
@karinazad karinazad deleted the ume-notebook-embeddings branch March 19, 2025 18:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants