Step 8000 (training checkpoint)
Browse files
README.md
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- llaza
|
| 4 |
+
- zip2zip
|
| 5 |
+
- adaptive-tokenization
|
| 6 |
+
library_name: zip2zip
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# zip2zip-phi35-ozzdata-v1
|
| 10 |
+
|
| 11 |
+
Training checkpoint from [zip2zip-core](https://github.com/epfl-dlab/zip2zip-core).
|
| 12 |
+
This is a **candidate** model (not production-ready).
|
| 13 |
+
|
| 14 |
+
## Training Config
|
| 15 |
+
|
| 16 |
+
| Field | Value |
|
| 17 |
+
|-------|-------|
|
| 18 |
+
| model_config | `Phi3.5-mini` |
|
| 19 |
+
| init_from | `microsoft/Phi-3.5-mini-instruct` |
|
| 20 |
+
| max_subtokens | 4 |
|
| 21 |
+
| max_codebook_size | 2048 |
|
| 22 |
+
| seq_len | 2048 |
|
| 23 |
+
| lr | 0.0003 |
|
| 24 |
+
| max_tokens | 262144000 |
|
| 25 |
+
| step | 8000 |
|
| 26 |
+
| data | `phi-1B-sft-ozzaligned` |
|
| 27 |
+
|
| 28 |
+
## Usage
|
| 29 |
+
|
| 30 |
+
This is a **training checkpoint** (torchtitan format). To use for inference,
|
| 31 |
+
export to HuggingFace format first:
|
| 32 |
+
|
| 33 |
+
```bash
|
| 34 |
+
python scripts/zip2zip_hf/export_to_zip2zip.py \
|
| 35 |
+
--ckpt_dir <local_path>/step_8000 \
|
| 36 |
+
--output_dir <export_dir> \
|
| 37 |
+
--base_model microsoft/Phi-3.5-mini-instruct \
|
| 38 |
+
--model_config Phi3.5-mini
|
| 39 |
+
```
|
meta.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f2525e2ae6941de3c422d78081f5f5f73eb96b92b4cd13b91f074c0e0c38b031
|
| 3 |
+
size 2843
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:df1812799ccd6f9834a9015b4a35c6eea1582a7770d7c694567b7347131eab59
|
| 3 |
+
size 16429773255
|
optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:21b7d45b0a318979c9049493cd4a14077fb796f0c17f52ac928827fb87e4eb7a
|
| 3 |
+
size 2290841064
|