How / where do I tell SaProt to use the fine tuned model?

[nogako]

I'm a newcomer in this field so my question my be trivial. But I'm trying to do mutation prediction using SaProt with the Thermostability fine tuned model. How or where do I tell SaProt to use the fine tuned model?

Thanks in advance.

[LTEnjoy]

Hi!

You can load the fine-tuned model through the from_checkpoint argument. e.g.,

config = {
    ...,
    "from_checkpoint": "your/path/to/weight.pt"
}

model = SaprotRegressionModel(**config)

[nogako]

Thank you very much. Where from do I import "SaprotRegressionModel"?

[LTEnjoy]

Please fitst git clone the newest repo, and then import SaprotRegressionModel by:

from model.saprot.saprot_regression_model import SaprotRegressionModel

[nogako]

Thank you