kparvataneni commited on
Commit
d85ec16
·
verified ·
1 Parent(s): a3b9ba5

Update model card with all SFT checkpoint revisions

Browse files
Files changed (1) hide show
  1. README.md +17 -7
README.md CHANGED
@@ -11,23 +11,33 @@ library_name: transformers
11
 
12
  # Llama 3.1 8B Instruct — balanced 3k SFT (synthetic canaries)
13
 
14
- Fine-tuned checkpoint from the **model-memo-diff** / PHIMemo project (PHI / benign / public seeded documents).
15
 
16
  ## Revisions
17
- | Revision | Training step | Role |
18
  |---|---|---|
19
- | `step-000160` | 160 | Early / low-memorization donor |
20
- | `step-000760` | 760 | Near-peak memorization |
 
 
 
 
 
 
 
 
 
21
 
22
  ## Load
23
  ```python
24
  from transformers import AutoModelForCausalLM, AutoTokenizer
25
  repo = "PHIMemo/llama31-8b-instruct-sft-balanced-3k"
26
- tok = AutoTokenizer.from_pretrained(repo, revision="step-000760")
27
- model = AutoModelForCausalLM.from_pretrained(repo, revision="step-000760", torch_dtype="auto", device_map="auto")
 
28
  ```
29
 
30
  ## Notes
31
  - Synthetic dossiers only (not real PHI).
32
- - Weights only (`model.safetensors`); optimizer / trainer state not uploaded.
33
  - Gated base model terms still apply for Llama 3.1.
 
11
 
12
  # Llama 3.1 8B Instruct — balanced 3k SFT (synthetic canaries)
13
 
14
+ Fine-tuned checkpoints from the **model-memo-diff** / PHIMemo project (PHI / benign / public seeded documents).
15
 
16
  ## Revisions
17
+ | Revision | Training step | Notes |
18
  |---|---|---|
19
+ | `step-000040` | 40 | SFT checkpoint |
20
+ | `step-000080` | 80 | SFT checkpoint |
21
+ | `step-000160` | 160 | early / donor |
22
+ | `step-000240` | 240 | SFT checkpoint |
23
+ | `step-000320` | 320 | SFT checkpoint |
24
+ | `step-000400` | 400 | SFT checkpoint |
25
+ | `step-000480` | 480 | SFT checkpoint |
26
+ | `step-000640` | 640 | SFT checkpoint |
27
+ | `step-000720` | 720 | SFT checkpoint |
28
+ | `step-000760` | 760 | near-peak |
29
+ | `step-000800` | 800 | near-peak |
30
 
31
  ## Load
32
  ```python
33
  from transformers import AutoModelForCausalLM, AutoTokenizer
34
  repo = "PHIMemo/llama31-8b-instruct-sft-balanced-3k"
35
+ rev = "step-000800" # or any revision above
36
+ tok = AutoTokenizer.from_pretrained(repo, revision=rev)
37
+ model = AutoModelForCausalLM.from_pretrained(repo, revision=rev, torch_dtype="auto", device_map="auto")
38
  ```
39
 
40
  ## Notes
41
  - Synthetic dossiers only (not real PHI).
42
+ - Weights only (`model.safetensors` + configs); optimizer / trainer state not uploaded.
43
  - Gated base model terms still apply for Llama 3.1.