Closed GuishePerez closed 1 month ago
This PR is related to the following issue https://github.com/BobXWu/FASTopic/issues/4.
Here we update the save_model() and load_model() methods by save() and from_pretrained() repectively.
save_model()
load_model()
save()
from_pretrained()
Before only pytorch model weights were saved and load. Now the FASTopic object itself is serialized along with its own model weights.
FASTopic
This PR is related to the following issue https://github.com/BobXWu/FASTopic/issues/4.
Here we update the
save_model()
andload_model()
methods bysave()
andfrom_pretrained()
repectively.Before only pytorch model weights were saved and load. Now the
FASTopic
object itself is serialized along with its own model weights.