search

BobXWu / FASTopic

A Fast, Adaptive, Stable, and Transferable Topic Model
https://pypi.org/project/fastopic/
Apache License 2.0
28 stars 3 forks source link

Update model save load methods #5

Closed GuishePerez closed 1 month ago

GuishePerez commented 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.

Before only pytorch model weights were saved and load. Now the FASTopic object itself is serialized along with its own model weights.