IBM/ModuleFormer Issues
torch.dtype is not respected
Closed 6Context length
Closed 3
ModuleFormer is a MoE-based architecture that includes two different types of experts: stick-breaking attention heads and feedforward experts. We released a collection of ModuleFormer-based Language Models (MoLM) ranging in scale from 4 billion to 8 billion parameters.