Inference Code Examples

#1
by AmelieSchreiber - opened

Could you please provide example inference code for this model? I have been having trouble getting the conda environment and the Dockerfile to work from the GitHub for MoleculeSTM, and it would be amazing to have a few examples of how to use the HuggingFace version of this model. This would also be great to have for ProteinDT.

Thanks in advance, and thank you for the awesome models!

Hi @AmelieSchreiber ,

Sure, let me check how to add such examples on HF. I am still learning to play around with HF :)

On the other hand, there is the notebook that includes inference for editing, in case it's helpful to you: https://github.com/chao1224/MoleculeSTM/blob/main/demos/demo_downstream_zero_shot_molecule_edit.ipynb

I also have the codes/checkpoints ready for ProteinDT, and I can show you how to use it if you email me.

Regards,
Shengchao

Email sent! Thank you!

Sign up or log in to comment