2.84 GB
gugarosa's picture
Adds support for MQA/GQA and attention mask during training / fine-tuning.
371fd51