About partitions in example.py

[Shi-YiWei]

Hello, I am currently experiencing two issues.

1. in example.py

forward_kwargs = {"partitions": model.make_partitions()}
model maybe not have function make_partitions()
it should be get _partitions()?

2. how to pass or define partitions

[radandreicristian]

Hi @Shi-YiWei , and sorry for the late response.

You are correct, there is an issue there.

I believe for Group Attention, you can make the partitions with model.set_partitions() when constructing the model, and then forward_kwargs = {"partitions": model.get_partitions()}

Feel free to create a pull request for this.