Downloads · 30 days
28
21% of all-time downloads
poc-pipeline-e2e-test
kshitijthakkar/poc-pipeline-e2e-test
poc-pipeline-e2e-test is a text generation model from kshitijthakkar. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
Custom Qwen3 MoE model trained with pipeline parallelism.
All-time downloads
134
Public
Parameters
39.4M
405 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors236 MB · 87%
Parameter types
How the weights are stored.
BF1639.4M · 100%
Try a prompt
Model card
From the Hugging Face model README
Qwen3 MoE (Mixture of Experts) — 39M Parameters
Custom Qwen3 MoE model trained with pipeline parallelism.
Model Details
| Property | Value |
|---|---|
| Total Parameters | 39,388,928 |
| Architecture | MoE (Mixture of Experts) |
| Hidden Size | 128 |
| Num Layers | 2 |
| Attention Heads | 4 |
| Context Length | 512 |
| Vocab Size | 151,936 |
| Num Experts | 4 |
| Top-K Experts | 2 |
| MoE Hidden Dim | 128 |
Evaluation Results
| Metric | Value |
|---|---|
| val_loss | 0.3217 |
| val_perplexity | 1.3795 |
| train_loss | 0.2418 |
| step | 2000 |
Usage
import torch
from safetensors.torch import load_file
# Load model weights
state_dict = load_file("model.safetensors")
Training
Trained using pipeline parallelism with the multi_gpu_pretraining framework.
- Task
- Text Generation
- Library
- transformers
- Type
- qwen3_moe
- License
- apache-2.0
- Created
- Feb 23, 2026
- Updated
- Feb 23, 2026