search

AI-HPC-Research-Team / GIT-Mol

A Multi-modal Large Language Model for Molecular Science with Graph, Image, and Text
MIT License
16 stars 1 forks source link

Inference Code #2

Open amelie-iska opened 3 months ago

amelie-iska commented 3 months ago

Thanks for the really cool work. I was just wondering if you could provide inference code for the model.

Pengfei-Liu-SYSU commented 2 months ago

I apologize for any inconvenience caused. The inference code and the finetuning training code are both located in the /train/finetune/molecule_translation.py file. You can refer to 'test_git' or use 'model.generate_text'.