utils.collators.mamba
utils.collators.mamba
collators for Mamba
Classes
| Name | Description |
|---|---|
| MambaDataCollator | Collator for State Space Models (Mamba) |
MambaDataCollator
utils.collators.mamba.MambaDataCollator(self, tokenizer)Collator for State Space Models (Mamba)