Collect all weights in a single task

[edervishaj]

I am trying to create a new merging method that requires access to a model's entire set of parameters (across all layers). My understanding is that currently each MergeTask operates on a single layer. Is it possible to override this flow within mergekit, for example through a class-based new merging task?

Thanks!