ganscs commited on
Commit
581a211
·
verified ·
1 Parent(s): 483864c

Validated standalone BF16 epoch 2, eight shards

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. epoch_index.json +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ import torch
40
  from transformers import AutoModelForCausalLM, AutoTokenizer
41
 
42
  model_id = "CompassioninMachineLearning/OLMo-3-7B-CPT-BF16"
43
- revision = "main"
44
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
45
  model = AutoModelForCausalLM.from_pretrained(
46
  model_id, revision=revision, dtype=torch.bfloat16, device_map="auto"
 
40
  from transformers import AutoModelForCausalLM, AutoTokenizer
41
 
42
  model_id = "CompassioninMachineLearning/OLMo-3-7B-CPT-BF16"
43
+ revision = "epoch-2"
44
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
45
  model = AutoModelForCausalLM.from_pretrained(
46
  model_id, revision=revision, dtype=torch.bfloat16, device_map="auto"
epoch_index.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "repo_id": "CompassioninMachineLearning/OLMo-3-7B-CPT-BF16",
3
  "default_epoch": 2,
4
- "this_revision": "main",
5
  "this_epoch": 2,
6
  "epochs": [
7
  {
 
1
  {
2
  "repo_id": "CompassioninMachineLearning/OLMo-3-7B-CPT-BF16",
3
  "default_epoch": 2,
4
+ "this_revision": "epoch-2",
5
  "this_epoch": 2,
6
  "epochs": [
7
  {