Hu commited on
Commit
9ff40ec
·
verified ·
1 Parent(s): d7e49e3

Document weight directories and correct pretrained model path

Browse files

Explain the standalone 1B-A0.75B pretraining release, clarify that root weights were removed, and provide subfolder loading instructions.

Files changed (1) hide show
  1. README.md +57 -15
README.md CHANGED
@@ -5,28 +5,70 @@ tags:
5
  - qwen4-exp
6
  - mixture-of-experts
7
  - language-model
8
- - randomly-initialized
9
  ---
10
 
11
- # FlashNext-MoE-9B-A1B
12
 
13
- Randomly initialized, language-only `Qwen4ExpForCausalLM` checkpoint.
 
 
14
 
15
- - Parameters: 9,000,082,560 (9.000082560B)
16
- - Approximate active backbone parameters per token: 0.995B
17
- - 32 layers: 24 Gated DeltaNet and 8 Qwen Sparse Attention
18
- - 64 routed experts, top-2, plus one shared expert per layer
19
- - No vision tower and no MTP head
20
- - BF16 weights in five safetensors shards
21
- - Qwen3.8-Flash-Next tokenizer and chat template
22
 
23
- These weights are random and must be pretrained or distilled before useful
24
- inference.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  ```python
 
27
  from transformers import AutoModelForCausalLM, AutoTokenizer
28
 
29
- path = "/path/to/flashnext-moe-9b/model"
30
- tokenizer = AutoTokenizer.from_pretrained(path)
31
- model = AutoModelForCausalLM.from_pretrained(path, dtype="bfloat16")
 
 
 
 
 
 
 
32
  ```
 
 
 
 
 
 
 
 
 
 
 
5
  - qwen4-exp
6
  - mixture-of-experts
7
  - language-model
8
+ - pretrained
9
  ---
10
 
11
+ # Modujo model weights
12
 
13
+ This repository keeps model variants in self-contained subdirectories. The
14
+ repository name is retained for compatibility, but the repository root no
15
+ longer contains model weights and must not be loaded directly.
16
 
17
+ ## Weight directories
 
 
 
 
 
 
18
 
19
+ | Path | Model | Meaning | Status |
20
+ | --- | --- | --- | --- |
21
+ | [`pretrain/Modujo-1B-A0.75B/`](./pretrain/Modujo-1B-A0.75B) | Modujo-1B-A0.75B | Continued-pretraining base model: 1.0038B total parameters and about 0.7490B active parameters per token | Available |
22
+ | Repository root | Historical 9B-A1B architecture metadata and shared tokenizer files | The original random-initialized 9B weight shards were removed; this is not a loadable model directory | No weights |
23
+
24
+ There are currently no SFT, chat, RL, or looped-model weights in this
25
+ repository. New variants should be published in their own named directories so
26
+ their training stage and actual parameter size remain explicit.
27
+
28
+ ## Pretrained base model
29
+
30
+ `pretrain/Modujo-1B-A0.75B/` is the released pretraining artifact.
31
+
32
+ - Architecture: `Qwen4ExpForCausalLM`
33
+ - Parameters: 1,003,821,539 total (1.0038B)
34
+ - Active parameters per token: approximately 749,017,571 (0.7490B)
35
+ - 36 layers with 8 routed experts per layer, top-2 routing, and one shared expert
36
+ - Attention layout: repeating 3 Gated DeltaNet layers + 1 dense-attention layer
37
+ - Context configuration: 32K maximum positions; training sequences were up to 2,048 tokens
38
+ - Weight format: BF16 safetensors
39
+ - Training stage: continued-pretraining base model
40
+ - Not instruction-tuned and not intended to be treated as a chat model
41
+ - No QSA sparse indexer in this release
42
+
43
+ Exact size metadata is recorded in
44
+ [`parameter_summary.json`](./pretrain/Modujo-1B-A0.75B/parameter_summary.json).
45
+
46
+ ## Loading
47
+
48
+ Pass the subdirectory explicitly:
49
 
50
  ```python
51
+ import torch
52
  from transformers import AutoModelForCausalLM, AutoTokenizer
53
 
54
+ repo_id = "Alexhu1999/Modujo-9B-A1B"
55
+ subfolder = "pretrain/Modujo-1B-A0.75B"
56
+
57
+ tokenizer = AutoTokenizer.from_pretrained(repo_id, subfolder=subfolder)
58
+ model = AutoModelForCausalLM.from_pretrained(
59
+ repo_id,
60
+ subfolder=subfolder,
61
+ torch_dtype=torch.bfloat16,
62
+ device_map="auto",
63
+ )
64
  ```
65
+
66
+ Loading only `Alexhu1999/Modujo-9B-A1B` without `subfolder` will fail because
67
+ there are intentionally no weights at the repository root.
68
+
69
+ ## Limitations
70
+
71
+ This is a base language model checkpoint. It can perform short text completion,
72
+ but long generations may repeat and instruction following is not yet stable.
73
+ Use a separately identified SFT or aligned release for assistant/chat use when
74
+ one becomes available.