Prashant Pandey commited on
Commit
f3ba1d2
·
verified ·
1 Parent(s): c42416f

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +42 -0
  2. config.json +17 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ library_name: pytorch
5
+ tags:
6
+ - deepseek
7
+ - moe
8
+ - mla
9
+ - tiny-stories
10
+ - text-generation
11
+ model_name: DeepSeekV3
12
+ base_model: deepseek-v3
13
+ ---
14
+
15
+ # DeepSeekV3
16
+
17
+ Implementation of the **DeepSeek-V3** architecture (8x2 MoE).
18
+
19
+ ## Architecture Details
20
+ - **Architecture**: Mixture of Experts (MoE) + Multi-Head Latent Attention (MLA)
21
+ - **Parameters**: ~196 million
22
+ - **Layers**: 6
23
+ - **Hidden Dimension**: 512
24
+ - **Experts**: 8 routed experts (Top-2) + 1 Shared Expert
25
+ - **Attention**: MLA (Multi-Head Latent Attention) with 8 heads and 64-dim Latent Space
26
+ - **Position Embeddings**: Sinusoidal
27
+
28
+ ## Training
29
+ - **Dataset**: TinyStories
30
+ - **Platform**: Kaggle (2x T4 GPUs)
31
+ - **Optimizer**: AdamW
32
+
33
+ ## How to use
34
+ This model uses a custom architecture implementation. To load it, you can use the `state_dict` found in `model.safetensors` alongside a loader.
35
+
36
+ ```python
37
+ from safetensors.torch import load_file
38
+ weights = load_file("model.safetensors")
39
+ ```
40
+
41
+ ## Dataset
42
+ Trained on the TinyStories dataset.
config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DeepseekV3ForCausalLM"
4
+ ],
5
+ "model_type": "deepseek_v3_mini",
6
+ "hidden_size": 512,
7
+ "intermediate_size": 1365,
8
+ "num_attention_heads": 8,
9
+ "num_hidden_layers": 6,
10
+ "n_routed_experts": 8,
11
+ "num_experts_per_tok": 2,
12
+ "latent_disturation_dim": 64,
13
+ "vocab_size": 50257,
14
+ "max_position_embeddings": 256,
15
+ "rms_norm_eps": 1e-06,
16
+ "torch_dtype": "float32"
17
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d74575a37e1b268088cb4925defe3405c43886773b2f8731c421c0615d3b69d
3
+ size 679343512