How to use Kush26/GPT2_small_GQA_MoE_DDP with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Kush26/GPT2_small_GQA_MoE_DDP", dtype="auto")
