How to inference with my own model?

[sudo1609]

After trained the model with my dataset. I got the checkpoints folder as shown below, but I don't know what to do next to be able to inference. Can you help me?

[linhduongtuan]

@TheK2NumberOne you can use the notebook (https://colab.research.google.com/drive/1LY5Ds6qyr_Drpp9WSdt-ZEMvvrFICdEx#scrollTo=VucO3HSMoJkz) or modify the generate_bloom.py, for instance.
Good luck

[TheNha]

Hi @linhduongtuan. I read the notebook and the generate_bloom.py, but I don't see how to load checkpoint after training.

@TheK2NumberOne Do you have any updates? Thank you for reply.