klemenk commited on
Commit
82130fb
·
verified ·
1 Parent(s): 39df650

Upload AuriStream checkpoint

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. config.json +5 -3
  3. model.safetensors +2 -2
README.md CHANGED
@@ -14,7 +14,7 @@ library_name: transformers
14
 
15
  This model predicts cochlear tokens from a tokenizer such as [WavCochCausalV8192](https://huggingface.co/TuKoResearch/WavCochCausalV8192).
16
 
17
- This repository contains a freshly initialized `AuriStream100M20PredConfig` model. The weights are random and have not been trained from a checkpoint. The model was instantiated through the native training model with seed `1110` before conversion to the Hugging Face format.
18
 
19
  ## Model Details
20
 
 
14
 
15
  This model predicts cochlear tokens from a tokenizer such as [WavCochCausalV8192](https://huggingface.co/TuKoResearch/WavCochCausalV8192).
16
 
17
+ Native training step-zero initialization for the 100M 20-prediction model. This exactly uses origin seed `11101994` and historical source commit `fad735c90bd44b4bcfd6bdf9429edbb640d2ebf5`, matching the start of [W&B run 3g8jxx2k](https://wandb.ai/NaudioGPT/auristream/runs/3g8jxx2k). The weights are untrained FP32 values produced before XLA/FSDP wrapping.
18
 
19
  ## Model Details
20
 
config.json CHANGED
@@ -8,7 +8,7 @@
8
  "AutoModel": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel",
9
  "AutoModelForCausalLM": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel"
10
  },
11
- "torch_dtype": "bfloat16",
12
  "transformers_version": "4.40.0",
13
  "vocab_size": 8192,
14
  "n_embd": 768,
@@ -18,6 +18,8 @@
18
  "dropout": 0.0,
19
  "bias": false,
20
  "input_conv_kernel_size": 0,
21
- "random_init_seed": 1110,
22
- "random_init_source": "auristream.auristream.model.AuriStream"
 
 
23
  }
 
8
  "AutoModel": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel",
9
  "AutoModelForCausalLM": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel"
10
  },
11
+ "torch_dtype": "float32",
12
  "transformers_version": "4.40.0",
13
  "vocab_size": 8192,
14
  "n_embd": 768,
 
18
  "dropout": 0.0,
19
  "bias": false,
20
  "input_conv_kernel_size": 0,
21
+ "random_init_seed": 11101994,
22
+ "random_init_source": "native-training-step-zero",
23
+ "random_init_source_commit": "fad735c90bd44b4bcfd6bdf9429edbb640d2ebf5",
24
+ "random_init_checkpoint_iteration": 0
25
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:adeb9808a1ad4578544bcbece627eddb2a9f991091f20a25f229f1e5a9add662
3
- size 434159608
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2d67219043868773f06dfcb001a680333e05260a05e0c8929ac33ca73f711e1
3
+ size 868309144